Searched refs:test_enqueue_with_weights (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/tests/ | ||
| D | tpu_embedding_v2_valid_input_test.py | 169 def test_enqueue_with_weights(self, ragged): member in TPUEmbeddingTest |
| D | tpu_embedding_v1_correctness_test.py | 268 def test_enqueue_with_weights(self, ragged): member in TPUEmbeddingV0CorrectnessTest |