Update ops-related pbtxt files.
PiperOrigin-RevId: 329389171 Change-Id: I3901bb842396b8ceb2e8d2f915cf84e066bda1cd
This commit is contained in:
parent
2238977c2a
commit
451030f130
@ -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"
|
||||
}
|
||||
}
|
||||
}
|
@ -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"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -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
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -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 {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user