Home
last modified time | relevance | path

Searched refs:aggregation_weights (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/ops/
Dtpu_ops.py322 aggregation_weights, argument
365 aggregation_weights=aggregation_weights,
379 aggregation_weights, argument
442 aggregation_weights=aggregation_weights,
459 aggregation_weights, argument
522 aggregation_weights=aggregation_weights,
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt8 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
Dapi_def_EnqueueTPUEmbeddingRaggedTensorBatch.pbtxt8 embedding_indices and aggregation_weights into rows.
21 name: "aggregation_weights"
63 embedding_indices and aggregation_weights.
68 sample_splits[i], embedding_indices[i] and aggregation_weights[i] correspond
73 embedding_indices and aggregation_weights, must have the same shape, i.e. rank 1
Dapi_def_EnqueueTPUEmbeddingSparseBatch.pbtxt8 feature to which the corresponding embedding_indices and aggregation_weights
21 name: "aggregation_weights"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DEnqueueTPUEmbeddingSparseTensorBatch.pbtxt14 name: "aggregation_weights"
62 name: "aggregation_weights"
149 name: "aggregation_weights"
244 name: "aggregation_weights"
DEnqueueTPUEmbeddingSparseBatch.pbtxt14 name: "aggregation_weights"
58 name: "aggregation_weights"
DEnqueueTPUEmbeddingRaggedTensorBatch.pbtxt14 name: "aggregation_weights"
109 name: "aggregation_weights"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DEnqueueTPUEmbeddingSparseTensorBatch.pbtxt14 name: "aggregation_weights"
62 name: "aggregation_weights"
149 name: "aggregation_weights"
DEnqueueTPUEmbeddingSparseBatch.pbtxt14 name: "aggregation_weights"
58 name: "aggregation_weights"
DEnqueueTPUEmbeddingRaggedTensorBatch.pbtxt14 name: "aggregation_weights"
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding.py191 aggregation_weights=None): argument
214 sample_indices, aggregation_weights)
221 aggregation_weights=weights.values if weights is not None else None)
233 aggregation_weights=None): argument
256 aggregation_weights)
263 aggregation_weights=weights.values if weights is not None else None)
1702 _check_agreement(enqueue_data.aggregation_weights,
1713 _check_agreement(enqueue_data.aggregation_weights,
1795 enqueue_data.aggregation_weights
1796 if enqueue_data.aggregation_weights is not None else float_zeros)
[all …]
Dtpu_embedding_v2.py1020 aggregation_weights=weights,
1029 aggregation_weights=weights,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td3744 sample_splits[i], embedding_indices[i] and aggregation_weights[i] correspond
3749 embedding_indices and aggregation_weights, must have the same shape, i.e. rank 1
3756 embedding_indices and aggregation_weights into rows.
3764 RaggedTensor.}]>:$aggregation_weights,
3804 feature to which the corresponding embedding_indices and aggregation_weights
3810 (training example, feature) -- aggregation weights.}]>:$aggregation_weights,
3834 sample_indices[i], embedding_indices[i] and aggregation_weights[i] correspond
3839 embedding_indices and aggregation_weights) must have the same shape, i.e. rank 1
3846 which the corresponding embedding_indices and aggregation_weights values
3852 embedding_lookup_sparse().}]>:$aggregation_weights,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt1349 …argspec: "args=[\'sample_splits\', \'embedding_indices\', \'aggregation_weights\', \'mode_override…
1353 …argspec: "args=[\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_overrid…
1357 …argspec: "args=[\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_overrid…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt1349 …argspec: "args=[\'sample_splits\', \'embedding_indices\', \'aggregation_weights\', \'mode_override…
1353 …argspec: "args=[\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_overrid…
1357 …argspec: "args=[\'sample_indices\', \'embedding_indices\', \'aggregation_weights\', \'mode_overrid…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go40494 …*Scope, sample_indices []tf.Output, embedding_indices []tf.Output, aggregation_weights []tf.Output…
40505 …t(sample_indices), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights), mode_over…
48796 …*Scope, sample_indices []tf.Output, embedding_indices []tf.Output, aggregation_weights []tf.Output…
48807 …t(sample_indices), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights), mode_over…
52102 … *Scope, sample_splits []tf.Output, embedding_indices []tf.Output, aggregation_weights []tf.Output…
52113 …st(sample_splits), tf.OutputList(embedding_indices), tf.OutputList(aggregation_weights), mode_over…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt14023 name: "aggregation_weights"
14126 name: "aggregation_weights"
14209 name: "aggregation_weights"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt22223 name: "aggregation_weights"
22267 name: "aggregation_weights"
22350 name: "aggregation_weights"
22398 name: "aggregation_weights"