blob: 71c98b7fd8120e7393249c6983e1b549b252713e [file] [log] [blame]
op {
name: "TopK"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "values"
type_attr: "T"
}
output_arg {
name: "indices"
type: DT_INT32
}
attr {
name: "k"
type: "int"
has_minimum: true
}
attr {
name: "sorted"
type: "bool"
default_value {
b: true
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_UINT16
type: DT_HALF
}
}
}
deprecation {
version: 7
}
}
op {
name: "TopK"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "values"
type_attr: "T"
}
output_arg {
name: "indices"
type: DT_INT32
}
attr {
name: "k"
type: "int"
has_minimum: true
}
attr {
name: "sorted"
type: "bool"
default_value {
b: true
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_UINT16
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
deprecation {
version: 7
}
}
op {
name: "TopK"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "values"
type_attr: "T"
}
output_arg {
name: "indices"
type: DT_INT32
}
attr {
name: "k"
type: "int"
has_minimum: true
}
attr {
name: "sorted"
type: "bool"
default_value {
b: true
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_INT64
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_UINT16
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BFLOAT16
}
}
}
deprecation {
version: 7
}
}
op {
name: "TopK"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "values"
type_attr: "T"
}
output_arg {
name: "indices"
type: DT_INT32
}
attr {
name: "k"
type: "int"
has_minimum: true
}
attr {
name: "sorted"
type: "bool"
default_value {
b: true
}
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_INT64
type: DT_BFLOAT16
type: DT_UINT16
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
}
}
}
deprecation {
version: 7
}
}