Searched defs:test_dim (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/test/onnx/ | ||
| H A D | test_operators.py | 1013 def test_dim(self): member in TestOperators |
| H A D | test_pytorch_onnx_onnxruntime.py | 8778 def test_dim(self): member in TestONNXRuntime |
| /aosp_15_r20/external/pytorch/test/fx/ | ||
| H A D | test_z3_gradual_types.py | 52 def test_dim(self): member in TorchDynamoUseCases |
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_sparse_csr.py | 950 def test_dim(self, layout): member in TestSparseCompressed |
| H A D | test_nestedtensor.py | 450 def test_dim(self): member in TestNestedTensor |
| H A D | test_jit.py | 692 def test_dim(): function |