Searched refs:EmbeddingModule (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/executorch/backends/vulkan/test/ |
| H A D | test_vulkan_delegate.py | 1616 class EmbeddingModule(torch.nn.Module): class
1625 EmbeddingModule(torch.nn.Embedding(5, 4)),
1631 class EmbeddingModule(torch.nn.Module): class
1640 EmbeddingModule(torch.nn.Embedding(5, 4)),
1646 class EmbeddingModule(torch.nn.Module): class
1655 EmbeddingModule(torch.nn.Embedding(5, 4)),
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_no_runtime.py | 670 class EmbeddingModule(torch.nn.Module): class
687 EmbeddingModule().eval(),
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_ptq.py | 47 EmbeddingModule,
771 model = EmbeddingModule().eval()
|
| /aosp_15_r20/external/executorch/backends/apple/mps/test/ |
| H A D | test_mps.py | 2480 class EmbeddingModule(torch.nn.Module): class
2491 EmbeddingModule(), model_inputs, func_name=inspect.stack()[0].function[5:]
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 2590 class EmbeddingModule(torch.nn.Module): class
2863 class EmbeddingModule(torch.nn.Module): class in TestHelperModules
|
| /aosp_15_r20/external/pytorch/test/quantization/pt2e/ |
| H A D | test_quantize_pt2e.py | 1589 m_eager = TestHelperModules.EmbeddingModule().eval()
|