diff options
author | 2017-05-09 10:48:59 -0700 | |
---|---|---|
committer | 2017-05-10 16:45:03 -0700 | |
commit | 04d0b083d1b3a663447b815c59ef94e35f3b3ed9 (patch) | |
tree | 2e44a6d1cc6d23007acd654a94d4a119eac947f7 | |
parent | d9fc16f61e89b12681b24dc4206370e5a74f4c6f (diff) |
Update ops-related pbtxt files.
PiperOrigin-RevId: 155524343
-rw-r--r-- | tensorflow/core/ops/compat/ops_history.v1.pbtxt | 93 | ||||
-rw-r--r-- | tensorflow/core/ops/ops.pbtxt | 105 |
2 files changed, 198 insertions, 0 deletions
diff --git a/tensorflow/core/ops/compat/ops_history.v1.pbtxt b/tensorflow/core/ops/compat/ops_history.v1.pbtxt index bc40715acc..d67cda2ae2 100644 --- a/tensorflow/core/ops/compat/ops_history.v1.pbtxt +++ b/tensorflow/core/ops/compat/ops_history.v1.pbtxt @@ -21572,6 +21572,99 @@ op { is_stateful: true } op { + name: "SparseCross" + 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" + } + output_arg { + name: "output_indices" + type: DT_INT64 + } + output_arg { + name: "output_values" + type_attr: "out_type" + } + output_arg { + name: "output_shape" + type: DT_INT64 + } + attr { + name: "N" + type: "int" + has_minimum: true + } + attr { + name: "hashed_output" + type: "bool" + } + attr { + name: "num_buckets" + type: "int" + has_minimum: true + } + attr { + name: "hash_key" + type: "int" + } + 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 + } + } + } + attr { + name: "out_type" + type: "type" + allowed_values { + list { + type: DT_INT64 + type: DT_STRING + } + } + } + attr { + name: "internal_type" + type: "type" + allowed_values { + list { + type: DT_INT64 + type: DT_STRING + } + } + } +} +op { name: "SparseDenseCwiseAdd" input_arg { name: "sp_indices" diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt index 06e88e8a52..7635bc84ee 100644 --- a/tensorflow/core/ops/ops.pbtxt +++ b/tensorflow/core/ops/ops.pbtxt @@ -22475,6 +22475,111 @@ op { is_stateful: true } op { + name: "SparseCross" + input_arg { + name: "indices" + description: "2-D. Indices of each input `SparseTensor`." + type: DT_INT64 + number_attr: "N" + } + input_arg { + name: "values" + description: "1-D. values of each `SparseTensor`." + type_list_attr: "sparse_types" + } + input_arg { + name: "shapes" + description: "1-D. Shapes of each `SparseTensor`." + type: DT_INT64 + number_attr: "N" + } + input_arg { + name: "dense_inputs" + description: "2-D. Columns represented by dense `Tensor`." + type_list_attr: "dense_types" + } + output_arg { + name: "output_indices" + description: "2-D. Indices of the concatenated `SparseTensor`." + type: DT_INT64 + } + output_arg { + name: "output_values" + description: "1-D. Non-empty values of the concatenated or hashed\n`SparseTensor`." + type_attr: "out_type" + } + output_arg { + name: "output_shape" + description: "1-D. Shape of the concatenated `SparseTensor`." + type: DT_INT64 + } + attr { + name: "N" + type: "int" + has_minimum: true + } + attr { + name: "hashed_output" + type: "bool" + description: "If true, returns the hash of the cross instead of the string.\nThis will allow us avoiding string manipulations." + } + attr { + name: "num_buckets" + type: "int" + description: "It is used if hashed_output is true.\noutput = hashed_value%num_buckets if num_buckets > 0 else hashed_value." + has_minimum: true + } + attr { + name: "hash_key" + type: "int" + description: "Specify the hash_key that will be used by the `FingerprintCat64`\nfunction to combine the crosses fingerprints." + } + 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 + } + } + } + attr { + name: "out_type" + type: "type" + allowed_values { + list { + type: DT_INT64 + type: DT_STRING + } + } + } + attr { + name: "internal_type" + type: "type" + allowed_values { + list { + type: DT_INT64 + type: DT_STRING + } + } + } + summary: "Generates sparse cross from a list of sparse and dense tensors." + description: "The op takes two lists, one of 2D `SparseTensor` and one of 2D `Tensor`, each\nrepresenting features of one feature column. It outputs a 2D `SparseTensor` with\nthe batchwise crosses of these features.\n\nFor example, if the inputs are\n\n inputs[0]: SparseTensor with shape = [2, 2]\n [0, 0]: \"a\"\n [1, 0]: \"b\"\n [1, 1]: \"c\"\n\n inputs[1]: SparseTensor with shape = [2, 1]\n [0, 0]: \"d\"\n [1, 0]: \"e\"\n\n inputs[2]: Tensor [[\"f\"], [\"g\"]]\n\nthen the output will be\n\n shape = [2, 2]\n [0, 0]: \"a_X_d_X_f\"\n [1, 0]: \"b_X_e_X_g\"\n [1, 1]: \"c_X_e_X_g\"\n\nif hashed_output=true then the output will be\n\n shape = [2, 2]\n [0, 0]: FingerprintCat64(\n Fingerprint64(\"f\"), FingerprintCat64(\n Fingerprint64(\"d\"), Fingerprint64(\"a\")))\n [1, 0]: FingerprintCat64(\n Fingerprint64(\"g\"), FingerprintCat64(\n Fingerprint64(\"e\"), Fingerprint64(\"b\")))\n [1, 1]: FingerprintCat64(\n Fingerprint64(\"g\"), FingerprintCat64(\n Fingerprint64(\"e\"), Fingerprint64(\"c\")))" +} +op { name: "SparseDenseCwiseAdd" input_arg { name: "sp_indices" |