Searched refs:custom_function_call_vmap_helper (Results 1 – 3 of 3) sorted by relevance
744 from torch._functorch.autograd_function import custom_function_call_vmap_helper755 return custom_function_call_vmap_helper(
315 return custom_function_call_vmap_helper(320 def custom_function_call_vmap_helper( function
1125 from torch._functorch.autograd_function import custom_function_call_vmap_helper1130 return custom_function_call_vmap_helper(