Home
last modified time | relevance | path

Searched refs:test_embedding (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/
H A Dtpu_embedding_v2_correctness_hd_sparse_forward_test.py28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
H A Dtpu_embedding_v2_correctness_ragged_forward_test.py28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
H A Dtpu_embedding_v2_correctness_hd_ragged_training_test.py28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
H A Dtpu_embedding_v2_correctness_hd_ragged_forward_test.py28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
H A Dtpu_embedding_v2_correctness_ragged_training_test.py28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
H A Dtpu_embedding_v2_correctness_sparse_training_test.py28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
H A Dtpu_embedding_v2_correctness_hd_sparse_training_test.py28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
H A Dtpu_embedding_v2_correctness_sparse_forward_test.py28 def test_embedding(self, optimizer_name): member in TPUEmbeddingCorrectnessTest
H A Dtpu_embedding_v1_correctness_test.py115 def test_embedding(self, optimizer_name, training, sparse, member in TPUEmbeddingV0CorrectnessTest
/aosp_15_r20/external/pytorch/test/mobile/
H A Dtest_quantize_fx_lite_script_module.py22 def test_embedding(self): member in TestLiteFuseFx
/aosp_15_r20/external/pytorch/test/
H A Dtest_tensorboard.py304 def test_embedding(self): member in TestTensorBoardEmbedding
H A Dtest_nestedtensor.py1018 def test_embedding(self, device, layout): member in TestNestedTensorDeviceType
/aosp_15_r20/external/pytorch/test/fx/
H A Dtest_z3_gradual_types.py724 def test_embedding(self): member in HFOperations
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_op.py4773 def test_embedding(self, num_embeddings, embedding_dim): member in TestQuantizedEmbeddingOps
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor.py4404 def test_embedding(self): member in CommonTemplate
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py8628 def test_embedding(self): member in TestQuantizeFxOps
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime.py10020 def test_embedding(self): member in TestONNXRuntime