Searched refs:sample_indices (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 317 def enqueue_tpu_embedding_sparse_batch(sample_indices, argument
360 sample_indices=sample_indices,
374 def enqueue_tpu_embedding_sparse_tensor_batch(sample_indices, argument
425 sample_indices=sample_indices,
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | resample.py | 95 sample_indices = _repeat_range(sample_counts)
97 sample_indices = array_ops.stop_gradient(sample_indices)
98 return [array_ops.gather(x, sample_indices) for x in inputs]
|
| /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
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 989 …argspec: "args=[\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_overrid…
993 …argspec: "args=[\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_overrid…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 989 …argspec: "args=[\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_overrid…
993 …argspec: "args=[\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_overrid…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 14065 func EnqueueTPUEmbeddingSparseTensorBatch(scope *Scope, sample_indices []tf.Output, embedding_indic…
14076 …tf.OutputList(sample_indices), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights…
25563 func EnqueueTPUEmbeddingSparseBatch(scope *Scope, sample_indices []tf.Output, embedding_indices []t…
25574 …tf.OutputList(sample_indices), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 22584 name: "sample_indices"
22628 name: "sample_indices"
22711 name: "sample_indices"
22759 name: "sample_indices"
|
| 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 | 10498 name: "sample_indices"
10581 name: "sample_indices"
|