Update ops-related pbtxt files.

PiperOrigin-RevId: 284282293
Change-Id: I87cc926089574e5688ee8bfaf6d569c5e6d791c8
This commit is contained in:
A. Unique TensorFlower 2019-12-06 16:08:46 -08:00 committed by TensorFlower Gardener
parent 6b4dc4e986
commit 8409e2a60e
9 changed files with 888 additions and 1 deletions

View File

@ -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
}

View File

@ -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
}
}
}

View File

@ -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"
}
}
}
}

View 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"
}
}
}
}

View File

@ -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"
}
}
}
}

View File

@ -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 {
}
}
}
}

View File

@ -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 {
}
}
}
}

View File

@ -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"
}
}
}
}

View File

@ -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 {