Home
last modified time | relevance | path

Searched refs:safe_embedding_lookup_sparse_v2 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dembedding_ops_test.py872 embedding_ops.safe_embedding_lookup_sparse_v2(embedding_weights,
888 embedding_ops.safe_embedding_lookup_sparse_v2(
904 embedding_ops.safe_embedding_lookup_sparse_v2(embedding_weights,
920 embedding_ops.safe_embedding_lookup_sparse_v2(embedding_weights,
952 embedding_ops.safe_embedding_lookup_sparse_v2(embedding_weights,
968 embedding_ops.safe_embedding_lookup_sparse_v2(
986 embedding_ops.safe_embedding_lookup_sparse_v2(embedding_weights,
1004 embedding_ops.safe_embedding_lookup_sparse_v2(embedding_weights,
/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py673 def safe_embedding_lookup_sparse_v2(embedding_weights, function
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable_test.py433 return embedding_ops.safe_embedding_lookup_sparse_v2(
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2.py1622 outputs.append(embedding_ops.safe_embedding_lookup_sparse_v2(