Home
last modified time | relevance | path

Searched refs:embedding_lookup_sparse (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_EnqueueTPUEmbeddingSparseTensorBatch.pbtxt9 belong. It corresponds to sp_ids.indices[:,0] in embedding_lookup_sparse().
16 It corresponds to sp_ids.values in embedding_lookup_sparse().
24 embedding_lookup_sparse().
64 summary: "Eases the porting of code that uses tf.nn.embedding_lookup_sparse()."
Dapi_def_EnqueueTPUEmbeddingSparseBatch.pbtxt56 This Op eases the porting of code that uses embedding_lookup_sparse(),
58 embedding_lookup_sparse() is required to produce the arguments to this Op,
/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py399 def embedding_lookup_sparse(params, function
667 return embedding_lookup_sparse(params, sp_ids, sp_weights, "div", name,
908 result = embedding_lookup_sparse(
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column.py2551 embedding_lookup_sparse = embedding_ops.safe_embedding_lookup_sparse
2554 embedding_lookup_sparse = embedding_ops.embedding_lookup_sparse_v2
2556 return embedding_lookup_sparse(
2701 embedding_lookup_sparse = embedding_ops.safe_embedding_lookup_sparse
2704 embedding_lookup_sparse = embedding_ops.embedding_lookup_sparse_v2
2706 return embedding_lookup_sparse(
Dfeature_column_v2.py2924 embedding_lookup_sparse = embedding_ops.safe_embedding_lookup_sparse
2927 embedding_lookup_sparse = embedding_ops.embedding_lookup_sparse_v2
2929 return embedding_lookup_sparse(
3223 embedding_lookup_sparse = embedding_ops.safe_embedding_lookup_sparse
3226 embedding_lookup_sparse = embedding_ops.embedding_lookup_sparse_v2
3228 return embedding_lookup_sparse(
/external/tensorflow/tensorflow/lite/kernels/
DAndroid.bp88 "embedding_lookup_sparse.cc",
DBUILD574 "embedding_lookup_sparse.cc",
/external/tensorflow/tensorflow/lite/
Dtflite_static.bp70 "kernels/embedding_lookup_sparse.cc",
/external/tensorflow/tensorflow/python/kernel_tests/
Dembedding_ops_test.py713 embedding_sum = embedding_ops.embedding_lookup_sparse(
764 y = embedding_ops.embedding_lookup_sparse(
791 embedding_ops.embedding_lookup_sparse(
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt152 name: "embedding_lookup_sparse"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt192 name: "embedding_lookup_sparse"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td3791 This Op eases the porting of code that uses embedding_lookup_sparse(),
3793 embedding_lookup_sparse() is required to produce the arguments to this Op,
3830 Eases the porting of code that uses tf.nn.embedding_lookup_sparse().
3847 belong. It corresponds to sp_ids.indices[:,0] in embedding_lookup_sparse().}]>:$sample_indices,
3849 It corresponds to sp_ids.values in embedding_lookup_sparse().}]>:$embedding_indices,
3852 embedding_lookup_sparse().}]>:$aggregation_weights,
/external/tensorflow/
DRELEASE.md3125 * Making ids unique in `nn.embedding_lookup_sparse`. This helps to reduce