Update ops-related pbtxt files.
PiperOrigin-RevId: 328940053 Change-Id: Id1ba8116f81d1cf863555844292c41c500759682
This commit is contained in:
parent
64a18194f5
commit
757befb73d
@ -260,81 +260,3 @@ 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,89 +369,3 @@ 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
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@ -23740,15 +23740,6 @@ op {
|
|||||||
list {
|
list {
|
||||||
s: "SAME"
|
s: "SAME"
|
||||||
s: "VALID"
|
s: "VALID"
|
||||||
s: "EXPLICIT"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
attr {
|
|
||||||
name: "explicit_paddings"
|
|
||||||
type: "list(int)"
|
|
||||||
default_value {
|
|
||||||
list {
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -24017,15 +24008,6 @@ op {
|
|||||||
list {
|
list {
|
||||||
s: "SAME"
|
s: "SAME"
|
||||||
s: "VALID"
|
s: "VALID"
|
||||||
s: "EXPLICIT"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
attr {
|
|
||||||
name: "explicit_paddings"
|
|
||||||
type: "list(int)"
|
|
||||||
default_value {
|
|
||||||
list {
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user