Update ops-related pbtxt files.
PiperOrigin-RevId: 168023895
This commit is contained in:
parent
cd68ce1f57
commit
c17096fe77
@ -13516,6 +13516,74 @@ 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_FLOAT
|
||||||
|
type: DT_DOUBLE
|
||||||
|
type: DT_INT32
|
||||||
|
type: DT_INT64
|
||||||
|
type: DT_UINT8
|
||||||
|
type: DT_INT16
|
||||||
|
type: DT_INT8
|
||||||
|
type: DT_UINT16
|
||||||
|
type: DT_HALF
|
||||||
|
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"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "data_format"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: "NHWC"
|
||||||
|
}
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
s: "NHWC"
|
||||||
|
s: "NCHW"
|
||||||
|
s: "NCHW_VECT_C"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
op {
|
op {
|
||||||
name: "MaxPool3D"
|
name: "MaxPool3D"
|
||||||
input_arg {
|
input_arg {
|
||||||
@ -14478,6 +14546,70 @@ op {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
op {
|
||||||
|
name: "MaxPoolV2"
|
||||||
|
input_arg {
|
||||||
|
name: "input"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "ksize"
|
||||||
|
type: DT_INT32
|
||||||
|
}
|
||||||
|
input_arg {
|
||||||
|
name: "strides"
|
||||||
|
type: DT_INT32
|
||||||
|
}
|
||||||
|
output_arg {
|
||||||
|
name: "output"
|
||||||
|
type_attr: "T"
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "T"
|
||||||
|
type: "type"
|
||||||
|
default_value {
|
||||||
|
type: DT_FLOAT
|
||||||
|
}
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
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_HALF
|
||||||
|
type: DT_QINT8
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "padding"
|
||||||
|
type: "string"
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
s: "SAME"
|
||||||
|
s: "VALID"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
attr {
|
||||||
|
name: "data_format"
|
||||||
|
type: "string"
|
||||||
|
default_value {
|
||||||
|
s: "NHWC"
|
||||||
|
}
|
||||||
|
allowed_values {
|
||||||
|
list {
|
||||||
|
s: "NHWC"
|
||||||
|
s: "NCHW"
|
||||||
|
s: "NCHW_VECT_C"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
op {
|
op {
|
||||||
name: "MaxPoolWithArgmax"
|
name: "MaxPoolWithArgmax"
|
||||||
input_arg {
|
input_arg {
|
||||||
|
@ -12629,6 +12629,7 @@ op {
|
|||||||
type: DT_INT8
|
type: DT_INT8
|
||||||
type: DT_UINT16
|
type: DT_UINT16
|
||||||
type: DT_HALF
|
type: DT_HALF
|
||||||
|
type: DT_QINT8
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -12668,6 +12669,7 @@ op {
|
|||||||
list {
|
list {
|
||||||
s: "NHWC"
|
s: "NHWC"
|
||||||
s: "NCHW"
|
s: "NCHW"
|
||||||
|
s: "NCHW_VECT_C"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -13402,6 +13404,7 @@ op {
|
|||||||
type: DT_INT8
|
type: DT_INT8
|
||||||
type: DT_UINT16
|
type: DT_UINT16
|
||||||
type: DT_HALF
|
type: DT_HALF
|
||||||
|
type: DT_QINT8
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -13427,6 +13430,7 @@ op {
|
|||||||
list {
|
list {
|
||||||
s: "NHWC"
|
s: "NHWC"
|
||||||
s: "NCHW"
|
s: "NCHW"
|
||||||
|
s: "NCHW_VECT_C"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user