Update ops-related pbtxt files.
PiperOrigin-RevId: 236706704
This commit is contained in:
parent
6d580992f6
commit
d79bd989b9
@ -37917,6 +37917,84 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "MaxPoolGradGradWithArgmax"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "grad"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "argmax"
|
||||
type_attr: "Targmax"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "ksize"
|
||||
type: "list(int)"
|
||||
has_minimum: true
|
||||
minimum: 4
|
||||
}
|
||||
attr {
|
||||
name: "strides"
|
||||
type: "list(int)"
|
||||
has_minimum: true
|
||||
minimum: 4
|
||||
}
|
||||
attr {
|
||||
name: "padding"
|
||||
type: "string"
|
||||
allowed_values {
|
||||
list {
|
||||
s: "SAME"
|
||||
s: "VALID"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "include_batch_in_index"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Targmax"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_INT32
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
}
|
||||
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
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "MaxPoolGradV2"
|
||||
input_arg {
|
||||
@ -38549,6 +38627,84 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "MaxPoolGradWithArgmax"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "grad"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "argmax"
|
||||
type_attr: "Targmax"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "ksize"
|
||||
type: "list(int)"
|
||||
has_minimum: true
|
||||
minimum: 4
|
||||
}
|
||||
attr {
|
||||
name: "strides"
|
||||
type: "list(int)"
|
||||
has_minimum: true
|
||||
minimum: 4
|
||||
}
|
||||
attr {
|
||||
name: "padding"
|
||||
type: "string"
|
||||
allowed_values {
|
||||
list {
|
||||
s: "SAME"
|
||||
s: "VALID"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "include_batch_in_index"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Targmax"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_INT32
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
}
|
||||
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
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "MaxPoolV2"
|
||||
input_arg {
|
||||
@ -39079,6 +39235,83 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "MaxPoolWithArgmax"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "argmax"
|
||||
type_attr: "Targmax"
|
||||
}
|
||||
attr {
|
||||
name: "ksize"
|
||||
type: "list(int)"
|
||||
has_minimum: true
|
||||
minimum: 4
|
||||
}
|
||||
attr {
|
||||
name: "strides"
|
||||
type: "list(int)"
|
||||
has_minimum: true
|
||||
minimum: 4
|
||||
}
|
||||
attr {
|
||||
name: "Targmax"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_INT64
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_INT32
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "padding"
|
||||
type: "string"
|
||||
allowed_values {
|
||||
list {
|
||||
s: "SAME"
|
||||
s: "VALID"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "include_batch_in_index"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
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
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "Maximum"
|
||||
input_arg {
|
||||
@ -79785,6 +80018,82 @@ op {
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "StatefulUniformFullInt"
|
||||
input_arg {
|
||||
name: "resource"
|
||||
type: DT_RESOURCE
|
||||
}
|
||||
input_arg {
|
||||
name: "algorithm"
|
||||
type: DT_INT64
|
||||
}
|
||||
input_arg {
|
||||
name: "shape"
|
||||
type_attr: "shape_dtype"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "dtype"
|
||||
}
|
||||
attr {
|
||||
name: "dtype"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_UINT64
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "shape_dtype"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "StatefulUniformInt"
|
||||
input_arg {
|
||||
name: "resource"
|
||||
type: DT_RESOURCE
|
||||
}
|
||||
input_arg {
|
||||
name: "algorithm"
|
||||
type: DT_INT64
|
||||
}
|
||||
input_arg {
|
||||
name: "shape"
|
||||
type_attr: "shape_dtype"
|
||||
}
|
||||
input_arg {
|
||||
name: "minval"
|
||||
type_attr: "dtype"
|
||||
}
|
||||
input_arg {
|
||||
name: "maxval"
|
||||
type_attr: "dtype"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "dtype"
|
||||
}
|
||||
attr {
|
||||
name: "dtype"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "shape_dtype"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "StatelessIf"
|
||||
input_arg {
|
||||
|
@ -19108,6 +19108,13 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "include_batch_in_index"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Targmax"
|
||||
type: "type"
|
||||
@ -19252,6 +19259,13 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "include_batch_in_index"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Targmax"
|
||||
type: "type"
|
||||
@ -19397,6 +19411,13 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "include_batch_in_index"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
@ -38307,6 +38328,82 @@ op {
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "StatefulUniformFullInt"
|
||||
input_arg {
|
||||
name: "resource"
|
||||
type: DT_RESOURCE
|
||||
}
|
||||
input_arg {
|
||||
name: "algorithm"
|
||||
type: DT_INT64
|
||||
}
|
||||
input_arg {
|
||||
name: "shape"
|
||||
type_attr: "shape_dtype"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "dtype"
|
||||
}
|
||||
attr {
|
||||
name: "dtype"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_UINT64
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "shape_dtype"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "StatefulUniformInt"
|
||||
input_arg {
|
||||
name: "resource"
|
||||
type: DT_RESOURCE
|
||||
}
|
||||
input_arg {
|
||||
name: "algorithm"
|
||||
type: DT_INT64
|
||||
}
|
||||
input_arg {
|
||||
name: "shape"
|
||||
type_attr: "shape_dtype"
|
||||
}
|
||||
input_arg {
|
||||
name: "minval"
|
||||
type_attr: "dtype"
|
||||
}
|
||||
input_arg {
|
||||
name: "maxval"
|
||||
type_attr: "dtype"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "dtype"
|
||||
}
|
||||
attr {
|
||||
name: "dtype"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "shape_dtype"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_INT64
|
||||
}
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "StatelessIf"
|
||||
input_arg {
|
||||
|
Loading…
Reference in New Issue
Block a user