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