Searched refs:sample_indices (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 320 def enqueue_tpu_embedding_sparse_batch(sample_indices, argument
363 sample_indices=sample_indices,
377 def enqueue_tpu_embedding_sparse_tensor_batch(sample_indices, argument
440 sample_indices=sample_indices,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 5 name: "sample_indices"
60 of the table_ids list must be equal to that of sample_indices,
66 sample_indices[i], embedding_indices[i] and aggregation_weights[i] correspond
70 The tensors at corresponding positions in the three input lists (sample_indices,
|
| D | api_def_EnqueueTPUEmbeddingSparseBatch.pbtxt | 5 name: "sample_indices"
9 values belong. sample_indices[i] must equal b * nf + f, where nf is the
|
| D | api_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 62 of the table_ids list must be equal to that of sample_indices,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 4 name: "sample_indices"
52 name: "sample_indices"
139 name: "sample_indices"
234 name: "sample_indices"
|
| D | EnqueueTPUEmbeddingSparseBatch.pbtxt | 4 name: "sample_indices"
48 name: "sample_indices"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 4 name: "sample_indices"
52 name: "sample_indices"
139 name: "sample_indices"
|
| D | EnqueueTPUEmbeddingSparseBatch.pbtxt | 4 name: "sample_indices"
48 name: "sample_indices"
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 190 sample_indices=None, argument
214 sample_indices, aggregation_weights)
1695 if enqueue_data.sample_indices is None and combiner:
1700 _check_agreement(enqueue_data.sample_indices, 'sample_indices',
1830 enqueue_data.sample_indices
1831 if enqueue_data.sample_indices is not None else int_zeros)
|
| D | tpu_embedding_v2.py | 1027 sample_indices=indices,
|
| /external/rust/crates/rand/ |
| D | CHANGELOG.md | 247 - Optimised and changed return type of the `sample_indices` function. (#479)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1353 …argspec: "args=[\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_overrid…
1357 …argspec: "args=[\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_overrid…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1353 …argspec: "args=[\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_overrid…
1357 …argspec: "args=[\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_overrid…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 3805 values belong. sample_indices[i] must equal b * nf + f, where nf is the
3807 b is in [0, batch size).}]>:$sample_indices,
3834 sample_indices[i], embedding_indices[i] and aggregation_weights[i] correspond
3838 The tensors at corresponding positions in the three input lists (sample_indices,
3847 belong. It corresponds to sp_ids.indices[:,0] in embedding_lookup_sparse().}]>:$sample_indices,
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 40494 func EnqueueTPUEmbeddingSparseBatch(scope *Scope, sample_indices []tf.Output, embedding_indices []t…
40505 …tf.OutputList(sample_indices), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights…
48796 func EnqueueTPUEmbeddingSparseTensorBatch(scope *Scope, sample_indices []tf.Output, embedding_indic…
48807 …tf.OutputList(sample_indices), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 22213 name: "sample_indices"
22257 name: "sample_indices"
22340 name: "sample_indices"
22388 name: "sample_indices"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 14116 name: "sample_indices"
14199 name: "sample_indices"
|