Searched refs:test_advanced_indexing (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/test/ | ||
| D | test_functionalization.py | 206 def test_advanced_indexing(self): member in TestFunctionalization |
| /external/pytorch/test/functorch/ | ||
| D | test_eager_transforms.py | 887 def test_advanced_indexing(self, device): member in TestGradTransform |
| D | test_vmap.py | 5086 def test_advanced_indexing(self, device): member in TestVmapOperatorsOpInfo |
| /external/pytorch/test/quantization/core/ | ||
| D | test_quantized_op.py | 2877 def test_advanced_indexing(self): member in TestQuantizedOps |