Searched refs:test_tensor_split (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_legacy_vmap.py | 2218 def test_tensor_split(self): member in TestVmapOperatorsLegacy |
| /aosp_15_r20/external/pytorch/test/dynamo/ | ||
| H A D | test_repros.py | 3706 def test_tensor_split(self): member in ReproTests |
| /aosp_15_r20/external/pytorch/test/functorch/ | ||
| H A D | test_vmap.py | 3024 def test_tensor_split(self): member in TestVmapOperators |
| /aosp_15_r20/external/pytorch/test/onnx/ | ||
| H A D | test_pytorch_onnx_onnxruntime.py | 6396 def test_tensor_split(self): member in TestONNXRuntime |