Searched refs:test_split_with_sizes (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/test/inductor/ | ||
| D | test_unbacked_symints.py | 93 def test_split_with_sizes(self, device): member in TestUnbackedSymints |
| /external/pytorch/test/ | ||
| D | test_functionalization.py | 856 def test_split_with_sizes(self): member in TestFunctionalization |
| D | test_nestedtensor.py | 3823 def test_split_with_sizes(self, device): member in TestNestedTensorSubclass |
| /external/pytorch/test/onnx/ | ||
| D | test_operators.py | 228 def test_split_with_sizes(self): member in TestOperators |