Searched refs:_wrap_method_with_vmap_fallback_check (Results 1 – 2 of 2) sorted by relevance
974 self._wrap_method_with_vmap_fallback_check(test_method),977 def _wrap_method_with_vmap_fallback_check(self, method): member in Namespace.TestVmapBaseLegacy1009 @self._wrap_method_with_vmap_fallback_check1018 @self._wrap_method_with_vmap_fallback_check
1407 self._wrap_method_with_vmap_fallback_check(test_method),1410 def _wrap_method_with_vmap_fallback_check(self, method): member in Namespace.TestVmapBase1441 @self._wrap_method_with_vmap_fallback_check1450 @self._wrap_method_with_vmap_fallback_check