Searched refs:table_ids (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 55 name: "table_ids"
59 corresponding input. The ith input is looked up using table_ids[i]. The size
60 of the table_ids list must be equal to that of sample_indices,
67 to the ith feature. table_ids[i] indicates which embedding table to look up ith
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 377 table_ids, argument
428 table_ids=table_ids,
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 651 sample_idcs, embedding_idcs, aggregation_weights, table_ids = (
657 table_ids,
672 sample_idcs, embedding_idcs, aggregation_weights, table_ids = (
685 table_ids.append(table_id)
687 return sample_idcs, embedding_idcs, aggregation_weights, table_ids
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 993 …', \'embedding_indices\', \'aggregation_weights\', \'mode_override\', \'table_ids\', \'device_ordi…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 993 …', \'embedding_indices\', \'aggregation_weights\', \'mode_override\', \'table_ids\', \'device_ordi…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 14065 … []tf.Output, aggregation_weights []tf.Output, mode_override tf.Output, table_ids []int64, optiona…
14069 attrs := map[string]interface{}{"table_ids": table_ids}
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 22380 name: "table_ids"
22467 name: "table_ids"
|
| D | ops_history.v1.pbtxt | 22751 name: "table_ids"
22838 name: "table_ids"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 10660 name: "table_ids"
|