Searched refs:aggregation_weights (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 319 aggregation_weights, argument
362 aggregation_weights=aggregation_weights,
376 aggregation_weights, argument
427 aggregation_weights=aggregation_weights,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt | 8 which the corresponding embedding_indices and aggregation_weights values
20 name: "aggregation_weights"
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
21 name: "aggregation_weights"
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding.py | 651 sample_idcs, embedding_idcs, aggregation_weights, table_ids = (
656 aggregation_weights,
672 sample_idcs, embedding_idcs, aggregation_weights, table_ids = (
684 aggregation_weights.append(array_ops.zeros([0]))
687 return sample_idcs, embedding_idcs, aggregation_weights, table_ids
|
| /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 …*Scope, sample_indices []tf.Output, embedding_indices []tf.Output, aggregation_weights []tf.Output…
14076 …t(sample_indices), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights), mode_over…
25563 …*Scope, sample_indices []tf.Output, embedding_indices []tf.Output, aggregation_weights []tf.Output…
25574 …t(sample_indices), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights), mode_over…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 22594 name: "aggregation_weights"
22638 name: "aggregation_weights"
22721 name: "aggregation_weights"
22769 name: "aggregation_weights"
|
| D | ops_history.v2.pbtxt | 22223 name: "aggregation_weights"
22267 name: "aggregation_weights"
22350 name: "aggregation_weights"
22398 name: "aggregation_weights"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 10508 name: "aggregation_weights"
10591 name: "aggregation_weights"
|