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 {
|
||||
name: "MaxPool3D"
|
||||
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 {
|
||||
name: "MaxPoolWithArgmax"
|
||||
input_arg {
|
||||
|
@ -12629,6 +12629,7 @@ op {
|
||||
type: DT_INT8
|
||||
type: DT_UINT16
|
||||
type: DT_HALF
|
||||
type: DT_QINT8
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -12668,6 +12669,7 @@ op {
|
||||
list {
|
||||
s: "NHWC"
|
||||
s: "NCHW"
|
||||
s: "NCHW_VECT_C"
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -13402,6 +13404,7 @@ op {
|
||||
type: DT_INT8
|
||||
type: DT_UINT16
|
||||
type: DT_HALF
|
||||
type: DT_QINT8
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -13427,6 +13430,7 @@ op {
|
||||
list {
|
||||
s: "NHWC"
|
||||
s: "NCHW"
|
||||
s: "NCHW_VECT_C"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user