blob: cdaeb5091c7b407addec2811bbf0cb79e61db2d2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
op {
graph_op_name: "BoostedTreesBucketize"
visibility: HIDDEN
in_arg {
name: "float_values"
description: <<END
float; List of Rank 2 Tensor each containing float values for a single feature.
END
}
in_arg {
name: "bucket_boundaries"
description: <<END
float; List of Rank 1 Tensors each containing the bucket boundaries for a single
feature.
END
}
out_arg {
name: "buckets"
description: <<END
int; List of Rank 2 Tensors each containing the bucketized values for a single feature.
END
}
attr {
name: "num_features"
description: <<END
inferred int; number of features.
END
}
summary: "Bucketize each feature based on bucket boundaries."
description: <<END
An op that returns a list of float tensors, where each tensor represents the
bucketized values for a single feature.
END
}
|
