Searched defs:_vmap_test (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/test/ |
| D | test_legacy_vmap.py | 904 def _vmap_test( function
1029 def _vmap_test(self, *args, **kwargs): member in TestVmapOperatorsLegacy
2545 def _vmap_test(self, *args, **kwargs): member in TestVmapBatchedGradientLegacy
|
| /external/pytorch/test/functorch/ |
| D | test_vmap.py | 1332 def _vmap_test( function
1466 def _vmap_test(self, *args, **kwargs): member in TestVmapOperators
3611 def _vmap_test(self, *args, **kwargs): member in TestVmapBatchedGradient
6142 def _vmap_test(self, *args, **kwargs): member in TestVmapDeviceType
6240 def _vmap_test(self, *args, **kwargs): member in TestVmapNestedTensor
|