Searched defs:test_arange (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/test/onnx/ | ||
| H A D | test_fx_to_onnx_with_onnxruntime.py | 481 def test_arange(self): member in TestFxToOnnxWithOnnxRuntime |
| /aosp_15_r20/external/executorch/exir/tests/ | ||
| H A D | test_passes.py | 1108 def test_arange(self) -> None: member in TestPasses |
| /aosp_15_r20/external/pytorch/test/fx/ | ||
| H A D | test_z3_gradual_types.py | 621 def test_arange(self): member in HFOperations |
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_tensor_creation_ops.py | 2345 def test_arange(self, device): member in TestTensorCreation |
| H A D | test_mps.py | 7731 def test_arange(self): member in TestMPS |