Home
last modified time | relevance | path

Searched refs:embedding_lookup_sparse_v2 (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_embedding_for_serving.py357 return embedding_ops.embedding_lookup_sparse_v2(
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dembedding_ops.py593 def embedding_lookup_sparse_v2(params, function
/aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/
H A Dfeature_column.py2550 embedding_lookup_sparse = embedding_ops.embedding_lookup_sparse_v2
2700 embedding_lookup_sparse = embedding_ops.embedding_lookup_sparse_v2
H A Dfeature_column_v2.py2994 embedding_lookup_sparse = embedding_ops.embedding_lookup_sparse_v2
3294 embedding_lookup_sparse = embedding_ops.embedding_lookup_sparse_v2
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dsharded_variable_test.py541 return embedding_ops.embedding_lookup_sparse_v2(sv, sp_ids, None)
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/
H A Dcore.py1236 outputs = embedding_ops.embedding_lookup_sparse_v2(