Update ops-related pbtxt files.

PiperOrigin-RevId: 236706704
This commit is contained in:
A. Unique TensorFlower 2019-03-04 12:50:07 -08:00 committed by TensorFlower Gardener
parent 6d580992f6
commit d79bd989b9
2 changed files with 406 additions and 0 deletions

View File

@ -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 {

View File

@ -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 {