Update ops-related pbtxt files.

PiperOrigin-RevId: 329389171
Change-Id: I3901bb842396b8ceb2e8d2f915cf84e066bda1cd
This commit is contained in:
A. Unique TensorFlower 2020-08-31 15:18:47 -07:00 committed by TensorFlower Gardener
parent 2238977c2a
commit 451030f130
4 changed files with 278 additions and 0 deletions

View File

@ -0,0 +1,48 @@
op {
name: "ImageProjectiveTransformV3"
input_arg {
name: "images"
type_attr: "dtype"
}
input_arg {
name: "transforms"
type: DT_FLOAT
}
input_arg {
name: "output_shape"
type: DT_INT32
}
input_arg {
name: "fill_value"
type: DT_FLOAT
}
output_arg {
name: "transformed_images"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
allowed_values {
list {
type: DT_UINT8
type: DT_INT32
type: DT_INT64
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "interpolation"
type: "string"
}
attr {
name: "fill_mode"
type: "string"
default_value {
s: "CONSTANT"
}
}
}

View File

@ -260,3 +260,81 @@ op {
}
}
}
op {
name: "MaxPool"
input_arg {
name: "input"
type_attr: "T"
}
output_arg {
name: "output"
type_attr: "T"
}
attr {
name: "T"
type: "type"
default_value {
type: DT_FLOAT
}
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
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_QINT8
}
}
}
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"
s: "EXPLICIT"
}
}
}
attr {
name: "explicit_paddings"
type: "list(int)"
default_value {
list {
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NHWC"
}
allowed_values {
list {
s: "NHWC"
s: "NCHW"
s: "NCHW_VECT_C"
}
}
}
}

View File

@ -369,3 +369,89 @@ op {
}
}
}
op {
name: "MaxPoolGrad"
input_arg {
name: "orig_input"
type_attr: "T"
}
input_arg {
name: "orig_output"
type_attr: "T"
}
input_arg {
name: "grad"
type_attr: "T"
}
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"
s: "EXPLICIT"
}
}
}
attr {
name: "explicit_paddings"
type: "list(int)"
default_value {
list {
}
}
}
attr {
name: "data_format"
type: "string"
default_value {
s: "NHWC"
}
allowed_values {
list {
s: "NHWC"
s: "NCHW"
}
}
}
attr {
name: "T"
type: "type"
default_value {
type: DT_FLOAT
}
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
}
}
}
}

View File

@ -19123,6 +19123,54 @@ op {
}
}
}
op {
name: "ImageProjectiveTransformV3"
input_arg {
name: "images"
type_attr: "dtype"
}
input_arg {
name: "transforms"
type: DT_FLOAT
}
input_arg {
name: "output_shape"
type: DT_INT32
}
input_arg {
name: "fill_value"
type: DT_FLOAT
}
output_arg {
name: "transformed_images"
type_attr: "dtype"
}
attr {
name: "dtype"
type: "type"
allowed_values {
list {
type: DT_UINT8
type: DT_INT32
type: DT_INT64
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "interpolation"
type: "string"
}
attr {
name: "fill_mode"
type: "string"
default_value {
s: "CONSTANT"
}
}
}
op {
name: "ImageSummary"
input_arg {
@ -23740,6 +23788,15 @@ op {
list {
s: "SAME"
s: "VALID"
s: "EXPLICIT"
}
}
}
attr {
name: "explicit_paddings"
type: "list(int)"
default_value {
list {
}
}
}
@ -24008,6 +24065,15 @@ op {
list {
s: "SAME"
s: "VALID"
s: "EXPLICIT"
}
}
}
attr {
name: "explicit_paddings"
type: "list(int)"
default_value {
list {
}
}
}