Update ops-related pbtxt files.
PiperOrigin-RevId: 306536433 Change-Id: I10ed89b9cc6983a8078e852afdbad6402d9fe6a7
This commit is contained in:
parent
ce55348ee9
commit
d9666eb32b
@ -308,3 +308,70 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "ArgMax"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "dimension"
|
||||
type_attr: "Tidx"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "output_type"
|
||||
}
|
||||
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_COMPLEX64
|
||||
type: DT_INT64
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_BFLOAT16
|
||||
type: DT_UINT16
|
||||
type: DT_COMPLEX128
|
||||
type: DT_HALF
|
||||
type: DT_UINT32
|
||||
type: DT_UINT64
|
||||
type: DT_BOOL
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Tidx"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_INT32
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_INT32
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "output_type"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_INT64
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_INT32
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -308,3 +308,70 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "ArgMin"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "dimension"
|
||||
type_attr: "Tidx"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "output_type"
|
||||
}
|
||||
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_COMPLEX64
|
||||
type: DT_INT64
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_BFLOAT16
|
||||
type: DT_UINT16
|
||||
type: DT_COMPLEX128
|
||||
type: DT_HALF
|
||||
type: DT_UINT32
|
||||
type: DT_UINT64
|
||||
type: DT_BOOL
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Tidx"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_INT32
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_INT32
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "output_type"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_INT64
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_INT32
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -2153,6 +2153,7 @@ op {
|
||||
type: DT_HALF
|
||||
type: DT_UINT32
|
||||
type: DT_UINT64
|
||||
type: DT_BOOL
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -2219,6 +2220,7 @@ op {
|
||||
type: DT_HALF
|
||||
type: DT_UINT32
|
||||
type: DT_UINT64
|
||||
type: DT_BOOL
|
||||
}
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user