Searched refs:test_repeat_interleave (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_proxy_tensor.py | 1076 def test_repeat_interleave(self): member in TestSymbolicTracing |
| H A D | test_tensor_creation_ops.py | 1159 def test_repeat_interleave(self, device): member in TestTensorCreation |
| H A D | test_torch.py | 2102 def test_repeat_interleave(self, device): member in TestTorchDeviceType |
| H A D | test_mps.py | 3838 def test_repeat_interleave(self, device="mps"): member in TestMPS |
| /aosp_15_r20/external/pytorch/test/inductor/ | ||
| H A D | test_cpu_repro.py | 705 def test_repeat_interleave(self): member in CPUReproTests |
| H A D | test_aot_inductor.py | 1361 def test_repeat_interleave(self): member in AOTInductorTestsTemplate |
| H A D | test_torchinductor.py | 4363 def test_repeat_interleave(self): member in CommonTemplate |
| /aosp_15_r20/external/pytorch/test/onnx/ | ||
| H A D | test_pytorch_onnx_onnxruntime.py | 5408 def test_repeat_interleave(self): member in TestONNXRuntime |