blob: abe818e666cafdbb82a86f194e4cf6f548d41bef [file] [log] [blame]
op {
name: "Bucketize"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type: DT_INT32
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_INT32
type: DT_INT64
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "boundaries"
type: "list(float)"
}
}