| /external/tensorflow/tensorflow/python/ops/ |
| D | embedding_ops.py | 400 sp_ids, argument
489 if not isinstance(sp_ids, sparse_tensor.SparseTensor):
495 sp_ids.values.get_shape().assert_is_compatible_with(
497 sp_ids.indices.get_shape().assert_is_compatible_with(
499 sp_ids.dense_shape.get_shape().assert_is_compatible_with(
505 params + [sp_ids]) as name:
506 segment_ids = sp_ids.indices[:, 0]
508 ids = sp_ids.values
583 sp_ids, argument
667 return embedding_lookup_sparse(params, sp_ids, sp_weights, "div", name,
|
| D | sparse_ops.py | 1794 def sparse_merge(sp_ids, sp_values, vocab_size, name=None, argument
1887 return sparse_merge_impl(sp_ids, sp_values, vocab_size, name, already_sorted)
1890 def sparse_merge_impl(sp_ids, argument
1896 if isinstance(sp_ids, sparse_tensor.SparseTensorValue) or isinstance(
1897 sp_ids, sparse_tensor.SparseTensor):
1898 sp_ids = [sp_ids]
1905 if not isinstance(sp_ids, collections_abc.Iterable):
1907 "Found %s" % type(sp_ids))
1916 if len(sp_ids) != len(vocab_size):
1919 with ops.name_scope(name, "SparseMerge", [sp_ids, sp_values]):
[all …]
|
| D | parsing_config.py | 703 sp_ids = tensor_dict[feature.index_key]
705 sp_ids = [tensor_dict[index_key] for index_key in feature.index_key]
708 sp_ids,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 9 belong. It corresponds to sp_ids.indices[:,0] in embedding_lookup_sparse().
16 It corresponds to sp_ids.values in embedding_lookup_sparse().
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | lookup_ops_test.py | 1143 sp_ids = table.lookup(sp_features)
1145 self.assertAllEqual([5], sp_ids.values._shape_as_list())
1148 [sp_ids.indices, sp_ids.values, sp_ids.dense_shape])
1191 sp_ids = table.lookup(sp_features)
1193 self.assertAllEqual([5], sp_ids.values._shape_as_list())
1196 [sp_ids.indices, sp_ids.values, sp_ids.dense_shape])
1237 sp_ids = table.lookup(sp_features)
1239 self.assertAllEqual([5], sp_ids.values._shape_as_list())
1242 [sp_ids.indices, sp_ids.values, sp_ids.dense_shape])
2836 sp_ids = table.lookup(sp_features)
[all …]
|
| D | embedding_ops_test.py | 671 sp_ids = sparse_tensor.SparseTensor(
680 return sp_ids, sp_weights, ids, weights, vals_per_batch_entry
697 sp_ids, sp_weights, ids, weights, vals_per_batch_entry = (
715 sp_ids,
754 sp_ids, sp_weights, _, _, _ = (self._RandomIdsAndWeights(
766 sp_ids,
781 sp_ids = sparse_tensor.SparseTensor(
792 x, sp_ids, sp_weights, combiner="mean")
|
| D | sparse_ops_test.py | 223 sp_ids=ids, sp_values=values, vocab_size=5)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable_test.py | 417 sp_ids = sparse_tensor.SparseTensor(
421 return embedding_ops.embedding_lookup_sparse_v2(sv, sp_ids, None)
425 sp_ids = sparse_tensor.SparseTensor(
434 sv, sp_ids, sp_weights)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.sparse.pbtxt | 61 …argspec: "args=[\'sp_ids\', \'sp_values\', \'vocab_size\', \'name\', \'already_sorted\'], varargs=…
|
| D | tensorflow.nn.pbtxt | 193 …argspec: "args=[\'params\', \'sp_ids\', \'sp_weights\', \'partition_strategy\', \'name\', \'combin…
|
| D | tensorflow.pbtxt | 2173 …argspec: "args=[\'sp_ids\', \'sp_values\', \'vocab_size\', \'name\', \'already_sorted\'], varargs=…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 153 …argspec: "args=[\'params\', \'sp_ids\', \'sp_weights\', \'combiner\', \'max_norm\', \'name\'], var…
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column.py | 3132 sp_ids=id_tensor,
|
| D | feature_column_v2.py | 4092 sp_ids=id_tensor, sp_values=weight_tensor, vocab_size=int(size))
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 3847 belong. It corresponds to sp_ids.indices[:,0] in embedding_lookup_sparse().}]>:$sample_indices,
3849 It corresponds to sp_ids.values in embedding_lookup_sparse().}]>:$embedding_indices,
|