diff --git a/tensorflow/core/ops/compat/ops_history_v2/DenseCountSparseOutput.pbtxt b/tensorflow/core/ops/compat/ops_history_v2/DenseCountSparseOutput.pbtxt index c5b845fd0fb..be566eab9f4 100644 --- a/tensorflow/core/ops/compat/ops_history_v2/DenseCountSparseOutput.pbtxt +++ b/tensorflow/core/ops/compat/ops_history_v2/DenseCountSparseOutput.pbtxt @@ -6,7 +6,7 @@ op { } input_arg { name: "weights" - type: DT_FLOAT + type_attr: "output_type" } output_arg { name: "output_indices" @@ -49,7 +49,7 @@ op { minimum: -1 } attr { - name: "binary_count" + name: "binary_output" type: "bool" } attr { @@ -57,8 +57,10 @@ op { type: "type" allowed_values { list { + type: DT_INT32 type: DT_INT64 type: DT_FLOAT + type: DT_DOUBLE } } } diff --git a/tensorflow/core/ops/compat/ops_history_v2/RaggedCountSparseOutput.pbtxt b/tensorflow/core/ops/compat/ops_history_v2/RaggedCountSparseOutput.pbtxt index 7f492418b48..aa1a4e07aaf 100644 --- a/tensorflow/core/ops/compat/ops_history_v2/RaggedCountSparseOutput.pbtxt +++ b/tensorflow/core/ops/compat/ops_history_v2/RaggedCountSparseOutput.pbtxt @@ -10,7 +10,7 @@ op { } input_arg { name: "weights" - type: DT_FLOAT + type_attr: "output_type" } output_arg { name: "output_indices" @@ -53,7 +53,7 @@ op { minimum: -1 } attr { - name: "binary_count" + name: "binary_output" type: "bool" } attr { @@ -61,8 +61,10 @@ op { type: "type" allowed_values { list { + type: DT_INT32 type: DT_INT64 type: DT_FLOAT + type: DT_DOUBLE } } } diff --git a/tensorflow/core/ops/compat/ops_history_v2/SparseCountSparseOutput.pbtxt b/tensorflow/core/ops/compat/ops_history_v2/SparseCountSparseOutput.pbtxt index b701e5fc0db..ed79733f97f 100644 --- a/tensorflow/core/ops/compat/ops_history_v2/SparseCountSparseOutput.pbtxt +++ b/tensorflow/core/ops/compat/ops_history_v2/SparseCountSparseOutput.pbtxt @@ -14,7 +14,7 @@ op { } input_arg { name: "weights" - type: DT_FLOAT + type_attr: "output_type" } output_arg { name: "output_indices" @@ -57,7 +57,7 @@ op { minimum: -1 } attr { - name: "binary_count" + name: "binary_output" type: "bool" } attr { @@ -65,8 +65,10 @@ op { type: "type" allowed_values { list { + type: DT_INT32 type: DT_INT64 type: DT_FLOAT + type: DT_DOUBLE } } } diff --git a/tensorflow/core/ops/compat/ops_history_v2/SparseCrossHashed.pbtxt b/tensorflow/core/ops/compat/ops_history_v2/SparseCrossHashed.pbtxt new file mode 100644 index 00000000000..73002a92f24 --- /dev/null +++ b/tensorflow/core/ops/compat/ops_history_v2/SparseCrossHashed.pbtxt @@ -0,0 +1,72 @@ +op { + name: "SparseCrossHashed" + input_arg { + name: "indices" + type: DT_INT64 + number_attr: "N" + } + input_arg { + name: "values" + type_list_attr: "sparse_types" + } + input_arg { + name: "shapes" + type: DT_INT64 + number_attr: "N" + } + input_arg { + name: "dense_inputs" + type_list_attr: "dense_types" + } + input_arg { + name: "num_buckets" + type: DT_INT64 + } + input_arg { + name: "strong_hash" + type: DT_BOOL + } + input_arg { + name: "salt" + type: DT_INT64 + } + output_arg { + name: "output_indices" + type: DT_INT64 + } + output_arg { + name: "output_values" + type: DT_INT64 + } + output_arg { + name: "output_shape" + type: DT_INT64 + } + attr { + name: "N" + type: "int" + has_minimum: true + } + attr { + name: "sparse_types" + type: "list(type)" + has_minimum: true + allowed_values { + list { + type: DT_INT64 + type: DT_STRING + } + } + } + attr { + name: "dense_types" + type: "list(type)" + has_minimum: true + allowed_values { + list { + type: DT_INT64 + type: DT_STRING + } + } + } +} diff --git a/tensorflow/core/ops/compat/ops_history_v2/SparseCrossV2.pbtxt b/tensorflow/core/ops/compat/ops_history_v2/SparseCrossV2.pbtxt new file mode 100644 index 00000000000..206542e4713 --- /dev/null +++ b/tensorflow/core/ops/compat/ops_history_v2/SparseCrossV2.pbtxt @@ -0,0 +1,64 @@ +op { + name: "SparseCrossV2" + input_arg { + name: "indices" + type: DT_INT64 + number_attr: "N" + } + input_arg { + name: "values" + type_list_attr: "sparse_types" + } + input_arg { + name: "shapes" + type: DT_INT64 + number_attr: "N" + } + input_arg { + name: "dense_inputs" + type_list_attr: "dense_types" + } + input_arg { + name: "sep" + type: DT_STRING + } + output_arg { + name: "output_indices" + type: DT_INT64 + } + output_arg { + name: "output_values" + type: DT_STRING + } + output_arg { + name: "output_shape" + type: DT_INT64 + } + attr { + name: "N" + type: "int" + has_minimum: true + } + attr { + name: "sparse_types" + type: "list(type)" + has_minimum: true + allowed_values { + list { + type: DT_INT64 + type: DT_STRING + } + } + } + attr { + name: "dense_types" + type: "list(type)" + has_minimum: true + allowed_values { + list { + type: DT_INT64 + type: DT_STRING + } + } + } +} diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt index 1ea06a2fdac..c951cb11778 100644 --- a/tensorflow/core/ops/ops.pbtxt +++ b/tensorflow/core/ops/ops.pbtxt @@ -11515,7 +11515,7 @@ op { } } attr { - name: "binary_count" + name: "binary_output" type: "bool" default_value { b: false @@ -11530,7 +11530,7 @@ op { } input_arg { name: "weights" - type: DT_FLOAT + type_attr: "output_type" } output_arg { name: "output_indices" @@ -11573,7 +11573,7 @@ op { minimum: -1 } attr { - name: "binary_count" + name: "binary_output" type: "bool" } attr { @@ -11581,8 +11581,10 @@ op { type: "type" allowed_values { list { + type: DT_INT32 type: DT_INT64 type: DT_FLOAT + type: DT_DOUBLE } } } @@ -33206,7 +33208,7 @@ op { } } attr { - name: "binary_count" + name: "binary_output" type: "bool" default_value { b: false @@ -33225,7 +33227,7 @@ op { } input_arg { name: "weights" - type: DT_FLOAT + type_attr: "output_type" } output_arg { name: "output_indices" @@ -33268,7 +33270,7 @@ op { minimum: -1 } attr { - name: "binary_count" + name: "binary_output" type: "bool" } attr { @@ -33276,8 +33278,10 @@ op { type: "type" allowed_values { list { + type: DT_INT32 type: DT_INT64 type: DT_FLOAT + type: DT_DOUBLE } } } @@ -44717,7 +44721,7 @@ op { } } attr { - name: "binary_count" + name: "binary_output" type: "bool" default_value { b: false @@ -44849,7 +44853,7 @@ op { } input_arg { name: "weights" - type: DT_FLOAT + type_attr: "output_type" } output_arg { name: "output_indices" @@ -44892,7 +44896,7 @@ op { minimum: -1 } attr { - name: "binary_count" + name: "binary_output" type: "bool" } attr { @@ -44900,8 +44904,10 @@ op { type: "type" allowed_values { list { + type: DT_INT32 type: DT_INT64 type: DT_FLOAT + type: DT_DOUBLE } } } @@ -44999,6 +45005,142 @@ op { } } } +op { + name: "SparseCrossHashed" + input_arg { + name: "indices" + type: DT_INT64 + number_attr: "N" + } + input_arg { + name: "values" + type_list_attr: "sparse_types" + } + input_arg { + name: "shapes" + type: DT_INT64 + number_attr: "N" + } + input_arg { + name: "dense_inputs" + type_list_attr: "dense_types" + } + input_arg { + name: "num_buckets" + type: DT_INT64 + } + input_arg { + name: "strong_hash" + type: DT_BOOL + } + input_arg { + name: "salt" + type: DT_INT64 + } + output_arg { + name: "output_indices" + type: DT_INT64 + } + output_arg { + name: "output_values" + type: DT_INT64 + } + output_arg { + name: "output_shape" + type: DT_INT64 + } + attr { + name: "N" + type: "int" + has_minimum: true + } + attr { + name: "sparse_types" + type: "list(type)" + has_minimum: true + allowed_values { + list { + type: DT_INT64 + type: DT_STRING + } + } + } + attr { + name: "dense_types" + type: "list(type)" + has_minimum: true + allowed_values { + list { + type: DT_INT64 + type: DT_STRING + } + } + } +} +op { + name: "SparseCrossV2" + input_arg { + name: "indices" + type: DT_INT64 + number_attr: "N" + } + input_arg { + name: "values" + type_list_attr: "sparse_types" + } + input_arg { + name: "shapes" + type: DT_INT64 + number_attr: "N" + } + input_arg { + name: "dense_inputs" + type_list_attr: "dense_types" + } + input_arg { + name: "sep" + type: DT_STRING + } + output_arg { + name: "output_indices" + type: DT_INT64 + } + output_arg { + name: "output_values" + type: DT_STRING + } + output_arg { + name: "output_shape" + type: DT_INT64 + } + attr { + name: "N" + type: "int" + has_minimum: true + } + attr { + name: "sparse_types" + type: "list(type)" + has_minimum: true + allowed_values { + list { + type: DT_INT64 + type: DT_STRING + } + } + } + attr { + name: "dense_types" + type: "list(type)" + has_minimum: true + allowed_values { + list { + type: DT_INT64 + type: DT_STRING + } + } + } +} op { name: "SparseDenseCwiseAdd" input_arg {