Searched refs:embedding_indices (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 318 embedding_indices, argument
361 embedding_indices=embedding_indices,
375 embedding_indices, argument
426 embedding_indices=embedding_indices,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 8 which the corresponding embedding_indices and aggregation_weights values
13 name: "embedding_indices"
61 embedding_indices and aggregation_weights.
66 sample_indices[i], embedding_indices[i] and aggregation_weights[i] correspond
71 embedding_indices and aggregation_weights) must have the same shape, i.e. rank 1
|
| D | api_def_EnqueueTPUEmbeddingSparseBatch.pbtxt | 8 feature to which the corresponding embedding_indices and aggregation_weights
15 name: "embedding_indices"
|
| /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 …mbeddingSparseTensorBatch(scope *Scope, sample_indices []tf.Output, embedding_indices []tf.Output,…
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 | 22589 name: "embedding_indices"
22633 name: "embedding_indices"
22716 name: "embedding_indices"
22764 name: "embedding_indices"
|
| D | ops_history.v2.pbtxt | 22218 name: "embedding_indices"
22262 name: "embedding_indices"
22345 name: "embedding_indices"
22393 name: "embedding_indices"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 10503 name: "embedding_indices"
10586 name: "embedding_indices"
|