/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/ |
H A D | tpu_embedding_v2_correctness_hd_sparse_forward_test.py | 28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
|
H A D | tpu_embedding_v2_correctness_ragged_forward_test.py | 28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
|
H A D | tpu_embedding_v2_correctness_hd_ragged_training_test.py | 28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
|
H A D | tpu_embedding_v2_correctness_hd_ragged_forward_test.py | 28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
|
H A D | tpu_embedding_v2_correctness_ragged_training_test.py | 28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
|
H A D | tpu_embedding_v2_correctness_sparse_training_test.py | 28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
|
H A D | tpu_embedding_v2_correctness_hd_sparse_training_test.py | 28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
|
H A D | tpu_embedding_v2_correctness_sparse_forward_test.py | 28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
|
H A D | tpu_embedding_v1_correctness_test.py | 115 def test_embedding(self, optimizer_name, training, sparse, member in TPUEmbeddingV0CorrectnessTest
|
/aosp_15_r20/external/pytorch/test/mobile/ |
H A D | test_quantize_fx_lite_script_module.py | 22 def test_embedding(self): member in TestLiteFuseFx
|
/aosp_15_r20/external/pytorch/test/ |
H A D | test_tensorboard.py | 304 def test_embedding(self): member in TestTensorBoardEmbedding
|
H A D | test_nestedtensor.py | 1018 def test_embedding(self, device, layout): member in TestNestedTensorDeviceType
|
/aosp_15_r20/external/pytorch/test/fx/ |
H A D | test_z3_gradual_types.py | 724 def test_embedding(self): member in HFOperations
|
/aosp_15_r20/external/pytorch/test/quantization/core/ |
H A D | test_quantized_op.py | 4773 def test_embedding(self, num_embeddings, embedding_dim): member in TestQuantizedEmbeddingOps
|
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_torchinductor.py | 4404 def test_embedding(self): member in CommonTemplate
|
/aosp_15_r20/external/pytorch/test/quantization/fx/ |
H A D | test_quantize_fx.py | 8628 def test_embedding(self): member in TestQuantizeFxOps
|
/aosp_15_r20/external/pytorch/test/onnx/ |
H A D | test_pytorch_onnx_onnxruntime.py | 10020 def test_embedding(self): member in TestONNXRuntime
|