Searched defs:test_numel (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_dynamic_shapes.py | 404 def test_numel(self): member in TestPySymInt |
| H A D | test_nestedtensor.py | 461 def test_numel(self): member in TestNestedTensor |
| H A D | test_torch.py | 9061 def test_numel(self): member in TestTorch |
| H A D | test_jit.py | 931 def test_numel(self): member in TestJit |
| /aosp_15_r20/external/pytorch/test/functorch/ | ||
| H A D | test_eager_transforms.py | 314 def test_numel(self, device): member in TestGradTransform |
| /aosp_15_r20/external/pytorch/test/dynamo/ | ||
| H A D | test_misc.py | 1698 def test_numel(self): member in MiscTests |
| /aosp_15_r20/external/pytorch/test/onnx/ | ||
| H A D | test_pytorch_onnx_onnxruntime.py | 7844 def test_numel(self): member in TestONNXRuntime |