Update ops-related pbtxt files.
PiperOrigin-RevId: 284282293 Change-Id: I87cc926089574e5688ee8bfaf6d569c5e6d791c8
This commit is contained in:
parent
6b4dc4e986
commit
8409e2a60e
@ -0,0 +1,28 @@
|
||||
op {
|
||||
name: "CheckNumericsV2"
|
||||
input_arg {
|
||||
name: "tensor"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_BFLOAT16
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "message"
|
||||
type: "string"
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
@ -27,3 +27,45 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "DebugNumericSummaryV2"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "output_dtype"
|
||||
}
|
||||
attr {
|
||||
name: "output_dtype"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_FLOAT
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
}
|
||||
attr {
|
||||
name: "tensor_debug_mode"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: -1
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "tensor_id"
|
||||
type: "int"
|
||||
default_value {
|
||||
i: -1
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@ -0,0 +1,38 @@
|
||||
op {
|
||||
name: "MatrixDiagPartV3"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "k"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "padding_value"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "diagonal"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
}
|
||||
attr {
|
||||
name: "align"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "RIGHT_LEFT"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "LEFT_RIGHT"
|
||||
s: "RIGHT_LEFT"
|
||||
s: "LEFT_LEFT"
|
||||
s: "RIGHT_RIGHT"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
46
tensorflow/core/ops/compat/ops_history_v1/MatrixDiagV3.pbtxt
Normal file
46
tensorflow/core/ops/compat/ops_history_v1/MatrixDiagV3.pbtxt
Normal file
@ -0,0 +1,46 @@
|
||||
op {
|
||||
name: "MatrixDiagV3"
|
||||
input_arg {
|
||||
name: "diagonal"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "k"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "num_rows"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "num_cols"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "padding_value"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
}
|
||||
attr {
|
||||
name: "align"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "RIGHT_LEFT"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "LEFT_RIGHT"
|
||||
s: "RIGHT_LEFT"
|
||||
s: "LEFT_LEFT"
|
||||
s: "RIGHT_RIGHT"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,38 @@
|
||||
op {
|
||||
name: "MatrixSetDiagV3"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "diagonal"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "k"
|
||||
type: DT_INT32
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
}
|
||||
attr {
|
||||
name: "align"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "RIGHT_LEFT"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "LEFT_RIGHT"
|
||||
s: "RIGHT_LEFT"
|
||||
s: "LEFT_LEFT"
|
||||
s: "RIGHT_RIGHT"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -109,3 +109,122 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "QuantizedDepthwiseConv2DWithBiasAndRelu"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "Tinput"
|
||||
}
|
||||
input_arg {
|
||||
name: "filter"
|
||||
type_attr: "Tfilter"
|
||||
}
|
||||
input_arg {
|
||||
name: "bias"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "min_input"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "max_input"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "min_filter"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "max_filter"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "out_type"
|
||||
}
|
||||
output_arg {
|
||||
name: "min_output"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
output_arg {
|
||||
name: "max_output"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
attr {
|
||||
name: "Tinput"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_QINT16
|
||||
type: DT_QUINT16
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Tfilter"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_QINT16
|
||||
type: DT_QUINT16
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "out_type"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_QINT32
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_QINT16
|
||||
type: DT_QUINT16
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "strides"
|
||||
type: "list(int)"
|
||||
}
|
||||
attr {
|
||||
name: "padding"
|
||||
type: "string"
|
||||
allowed_values {
|
||||
list {
|
||||
s: "SAME"
|
||||
s: "VALID"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "dilations"
|
||||
type: "list(int)"
|
||||
default_value {
|
||||
list {
|
||||
i: 1
|
||||
i: 1
|
||||
i: 1
|
||||
i: 1
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "padding_list"
|
||||
type: "list(int)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@ -127,3 +127,140 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "Tinput"
|
||||
}
|
||||
input_arg {
|
||||
name: "filter"
|
||||
type_attr: "Tfilter"
|
||||
}
|
||||
input_arg {
|
||||
name: "bias"
|
||||
type_attr: "Tbias"
|
||||
}
|
||||
input_arg {
|
||||
name: "min_input"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "max_input"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "min_filter"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "max_filter"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "min_freezed_output"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "max_freezed_output"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "out_type"
|
||||
}
|
||||
output_arg {
|
||||
name: "min_output"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
output_arg {
|
||||
name: "max_output"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
attr {
|
||||
name: "Tinput"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_QINT16
|
||||
type: DT_QUINT16
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Tfilter"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_QINT16
|
||||
type: DT_QUINT16
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Tbias"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_FLOAT
|
||||
type: DT_QINT32
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "out_type"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_QUINT8
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_QINT16
|
||||
type: DT_QUINT16
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "strides"
|
||||
type: "list(int)"
|
||||
}
|
||||
attr {
|
||||
name: "padding"
|
||||
type: "string"
|
||||
allowed_values {
|
||||
list {
|
||||
s: "SAME"
|
||||
s: "VALID"
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "dilations"
|
||||
type: "list(int)"
|
||||
default_value {
|
||||
list {
|
||||
i: 1
|
||||
i: 1
|
||||
i: 1
|
||||
i: 1
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "padding_list"
|
||||
type: "list(int)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@ -0,0 +1,130 @@
|
||||
op {
|
||||
name: "QuantizedMatMulWithBiasAndRequantize"
|
||||
input_arg {
|
||||
name: "a"
|
||||
type_attr: "T1"
|
||||
}
|
||||
input_arg {
|
||||
name: "b"
|
||||
type_attr: "T2"
|
||||
}
|
||||
input_arg {
|
||||
name: "bias"
|
||||
type_attr: "Tbias"
|
||||
}
|
||||
input_arg {
|
||||
name: "min_a"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "max_a"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "min_b"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "max_b"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "min_freezed_output"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "max_freezed_output"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
output_arg {
|
||||
name: "out"
|
||||
type_attr: "Toutput"
|
||||
}
|
||||
output_arg {
|
||||
name: "min_out"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
output_arg {
|
||||
name: "max_out"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
attr {
|
||||
name: "T1"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_QINT16
|
||||
type: DT_QUINT16
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "T2"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_QINT16
|
||||
type: DT_QUINT16
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Tbias"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_FLOAT
|
||||
type: DT_QINT32
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Toutput"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_QUINT8
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_QINT16
|
||||
type: DT_QUINT16
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "transpose_a"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "transpose_b"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "input_quant_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "MIN_FIRST"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "MIN_FIRST"
|
||||
s: "SCALED"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -6598,6 +6598,34 @@ op {
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "CheckNumericsV2"
|
||||
input_arg {
|
||||
name: "tensor"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_BFLOAT16
|
||||
type: DT_HALF
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "message"
|
||||
type: "string"
|
||||
}
|
||||
is_stateful: true
|
||||
}
|
||||
op {
|
||||
name: "Cholesky"
|
||||
input_arg {
|
||||
@ -10611,7 +10639,20 @@ op {
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type: DT_FLOAT
|
||||
type_attr: "output_dtype"
|
||||
}
|
||||
attr {
|
||||
name: "output_dtype"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_FLOAT
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_FLOAT
|
||||
type: DT_DOUBLE
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
@ -21523,6 +21564,44 @@ op {
|
||||
type: "type"
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "MatrixDiagPartV3"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "k"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "padding_value"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "diagonal"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
}
|
||||
attr {
|
||||
name: "align"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "RIGHT_LEFT"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "LEFT_RIGHT"
|
||||
s: "RIGHT_LEFT"
|
||||
s: "LEFT_LEFT"
|
||||
s: "RIGHT_RIGHT"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "MatrixDiagV2"
|
||||
input_arg {
|
||||
@ -21554,6 +21633,52 @@ op {
|
||||
type: "type"
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "MatrixDiagV3"
|
||||
input_arg {
|
||||
name: "diagonal"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "k"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "num_rows"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "num_cols"
|
||||
type: DT_INT32
|
||||
}
|
||||
input_arg {
|
||||
name: "padding_value"
|
||||
type_attr: "T"
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
}
|
||||
attr {
|
||||
name: "align"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "RIGHT_LEFT"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "LEFT_RIGHT"
|
||||
s: "RIGHT_LEFT"
|
||||
s: "LEFT_LEFT"
|
||||
s: "RIGHT_RIGHT"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "MatrixExponential"
|
||||
input_arg {
|
||||
@ -21676,6 +21801,44 @@ op {
|
||||
type: "type"
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "MatrixSetDiagV3"
|
||||
input_arg {
|
||||
name: "input"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "diagonal"
|
||||
type_attr: "T"
|
||||
}
|
||||
input_arg {
|
||||
name: "k"
|
||||
type: DT_INT32
|
||||
}
|
||||
output_arg {
|
||||
name: "output"
|
||||
type_attr: "T"
|
||||
}
|
||||
attr {
|
||||
name: "T"
|
||||
type: "type"
|
||||
}
|
||||
attr {
|
||||
name: "align"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "RIGHT_LEFT"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "LEFT_RIGHT"
|
||||
s: "RIGHT_LEFT"
|
||||
s: "LEFT_LEFT"
|
||||
s: "RIGHT_RIGHT"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "MatrixSolve"
|
||||
input_arg {
|
||||
@ -29971,6 +30134,14 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "padding_list"
|
||||
type: "list(int)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize"
|
||||
@ -30100,6 +30271,14 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "padding_list"
|
||||
type: "list(int)"
|
||||
default_value {
|
||||
list {
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "QuantizedInstanceNorm"
|
||||
@ -30651,6 +30830,136 @@ op {
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "QuantizedMatMulWithBiasAndRequantize"
|
||||
input_arg {
|
||||
name: "a"
|
||||
type_attr: "T1"
|
||||
}
|
||||
input_arg {
|
||||
name: "b"
|
||||
type_attr: "T2"
|
||||
}
|
||||
input_arg {
|
||||
name: "bias"
|
||||
type_attr: "Tbias"
|
||||
}
|
||||
input_arg {
|
||||
name: "min_a"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "max_a"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "min_b"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "max_b"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "min_freezed_output"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
input_arg {
|
||||
name: "max_freezed_output"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
output_arg {
|
||||
name: "out"
|
||||
type_attr: "Toutput"
|
||||
}
|
||||
output_arg {
|
||||
name: "min_out"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
output_arg {
|
||||
name: "max_out"
|
||||
type: DT_FLOAT
|
||||
}
|
||||
attr {
|
||||
name: "T1"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_QINT16
|
||||
type: DT_QUINT16
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "T2"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_QINT16
|
||||
type: DT_QUINT16
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Tbias"
|
||||
type: "type"
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_FLOAT
|
||||
type: DT_QINT32
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "Toutput"
|
||||
type: "type"
|
||||
default_value {
|
||||
type: DT_QUINT8
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
type: DT_QINT8
|
||||
type: DT_QUINT8
|
||||
type: DT_QINT32
|
||||
type: DT_QINT16
|
||||
type: DT_QUINT16
|
||||
}
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "transpose_a"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "transpose_b"
|
||||
type: "bool"
|
||||
default_value {
|
||||
b: false
|
||||
}
|
||||
}
|
||||
attr {
|
||||
name: "input_quant_mode"
|
||||
type: "string"
|
||||
default_value {
|
||||
s: "MIN_FIRST"
|
||||
}
|
||||
allowed_values {
|
||||
list {
|
||||
s: "MIN_FIRST"
|
||||
s: "SCALED"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
op {
|
||||
name: "QuantizedMaxPool"
|
||||
input_arg {
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user