From 8409e2a60e498213d9a745cdcfed996306df1996 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 6 Dec 2019 16:08:46 -0800 Subject: [PATCH] Update ops-related pbtxt files. PiperOrigin-RevId: 284282293 Change-Id: I87cc926089574e5688ee8bfaf6d569c5e6d791c8 --- .../ops_history_v1/CheckNumericsV2.pbtxt | 28 ++ .../DebugNumericSummaryV2.pbtxt | 42 +++ .../ops_history_v1/MatrixDiagPartV3.pbtxt | 38 +++ .../compat/ops_history_v1/MatrixDiagV3.pbtxt | 46 +++ .../ops_history_v1/MatrixSetDiagV3.pbtxt | 38 +++ ...ntizedDepthwiseConv2DWithBiasAndRelu.pbtxt | 119 +++++++ ...seConv2DWithBiasAndReluAndRequantize.pbtxt | 137 ++++++++ ...QuantizedMatMulWithBiasAndRequantize.pbtxt | 130 ++++++++ tensorflow/core/ops/ops.pbtxt | 311 +++++++++++++++++- 9 files changed, 888 insertions(+), 1 deletion(-) create mode 100644 tensorflow/core/ops/compat/ops_history_v1/CheckNumericsV2.pbtxt create mode 100644 tensorflow/core/ops/compat/ops_history_v1/MatrixDiagPartV3.pbtxt create mode 100644 tensorflow/core/ops/compat/ops_history_v1/MatrixDiagV3.pbtxt create mode 100644 tensorflow/core/ops/compat/ops_history_v1/MatrixSetDiagV3.pbtxt create mode 100644 tensorflow/core/ops/compat/ops_history_v1/QuantizedMatMulWithBiasAndRequantize.pbtxt diff --git a/tensorflow/core/ops/compat/ops_history_v1/CheckNumericsV2.pbtxt b/tensorflow/core/ops/compat/ops_history_v1/CheckNumericsV2.pbtxt new file mode 100644 index 00000000000..ba06f6a3524 --- /dev/null +++ b/tensorflow/core/ops/compat/ops_history_v1/CheckNumericsV2.pbtxt @@ -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 +} diff --git a/tensorflow/core/ops/compat/ops_history_v1/DebugNumericSummaryV2.pbtxt b/tensorflow/core/ops/compat/ops_history_v1/DebugNumericSummaryV2.pbtxt index d55c9678e7e..31f7efec2ab 100644 --- a/tensorflow/core/ops/compat/ops_history_v1/DebugNumericSummaryV2.pbtxt +++ b/tensorflow/core/ops/compat/ops_history_v1/DebugNumericSummaryV2.pbtxt @@ -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 + } + } +} diff --git a/tensorflow/core/ops/compat/ops_history_v1/MatrixDiagPartV3.pbtxt b/tensorflow/core/ops/compat/ops_history_v1/MatrixDiagPartV3.pbtxt new file mode 100644 index 00000000000..75a1307f279 --- /dev/null +++ b/tensorflow/core/ops/compat/ops_history_v1/MatrixDiagPartV3.pbtxt @@ -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" + } + } + } +} diff --git a/tensorflow/core/ops/compat/ops_history_v1/MatrixDiagV3.pbtxt b/tensorflow/core/ops/compat/ops_history_v1/MatrixDiagV3.pbtxt new file mode 100644 index 00000000000..793efcec3b6 --- /dev/null +++ b/tensorflow/core/ops/compat/ops_history_v1/MatrixDiagV3.pbtxt @@ -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" + } + } + } +} diff --git a/tensorflow/core/ops/compat/ops_history_v1/MatrixSetDiagV3.pbtxt b/tensorflow/core/ops/compat/ops_history_v1/MatrixSetDiagV3.pbtxt new file mode 100644 index 00000000000..d5a6af8b119 --- /dev/null +++ b/tensorflow/core/ops/compat/ops_history_v1/MatrixSetDiagV3.pbtxt @@ -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" + } + } + } +} diff --git a/tensorflow/core/ops/compat/ops_history_v1/QuantizedDepthwiseConv2DWithBiasAndRelu.pbtxt b/tensorflow/core/ops/compat/ops_history_v1/QuantizedDepthwiseConv2DWithBiasAndRelu.pbtxt index cc6d92389f7..8265ccb381c 100644 --- a/tensorflow/core/ops/compat/ops_history_v1/QuantizedDepthwiseConv2DWithBiasAndRelu.pbtxt +++ b/tensorflow/core/ops/compat/ops_history_v1/QuantizedDepthwiseConv2DWithBiasAndRelu.pbtxt @@ -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 { + } + } + } +} diff --git a/tensorflow/core/ops/compat/ops_history_v1/QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt b/tensorflow/core/ops/compat/ops_history_v1/QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt index 5413d151a6c..43d2e5d9036 100644 --- a/tensorflow/core/ops/compat/ops_history_v1/QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt +++ b/tensorflow/core/ops/compat/ops_history_v1/QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize.pbtxt @@ -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 { + } + } + } +} diff --git a/tensorflow/core/ops/compat/ops_history_v1/QuantizedMatMulWithBiasAndRequantize.pbtxt b/tensorflow/core/ops/compat/ops_history_v1/QuantizedMatMulWithBiasAndRequantize.pbtxt new file mode 100644 index 00000000000..1aab9762eea --- /dev/null +++ b/tensorflow/core/ops/compat/ops_history_v1/QuantizedMatMulWithBiasAndRequantize.pbtxt @@ -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" + } + } + } +} diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt index a9f479d5b65..6f07a4b91c1 100644 --- a/tensorflow/core/ops/ops.pbtxt +++ b/tensorflow/core/ops/ops.pbtxt @@ -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 {