Searched refs:sample_splits (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 457 def enqueue_tpu_embedding_ragged_tensor_batch(sample_splits, argument
520 sample_splits=sample_splits,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 4 name: "sample_splits"
99 name: "sample_splits"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 5 name: "sample_splits"
68 sample_splits[i], embedding_indices[i] and aggregation_weights[i] correspond
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt | 4 name: "sample_splits"
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 232 sample_splits=None, argument
255 cls).__new__(cls, embedding_indices, sample_splits,
1706 if enqueue_data.sample_splits is None and combiner:
1711 _check_agreement(enqueue_data.sample_splits, 'sample_splits', feature,
1791 enqueue_data.sample_splits
1792 if enqueue_data.sample_splits is not None else int_zeros)
|
| D | tpu_embedding_v2.py | 1018 sample_splits=indices,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 1349 …argspec: "args=[\'sample_splits\', \'embedding_indices\', \'aggregation_weights\', \'mode_override…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1349 …argspec: "args=[\'sample_splits\', \'embedding_indices\', \'aggregation_weights\', \'mode_override…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 52102 func EnqueueTPUEmbeddingRaggedTensorBatch(scope *Scope, sample_splits []tf.Output, embedding_indice…
52113 …tf.OutputList(sample_splits), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights)…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 3744 sample_splits[i], embedding_indices[i] and aggregation_weights[i] correspond
3758 RaggedTensor.}]>:$sample_splits,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 14013 name: "sample_splits"
|