Searched refs:custom_function_call (Results 1 – 2 of 2) sorted by relevance
58 custom_function_call = CustomFunctionHigherOrderOperator() variable88 @custom_function_call.py_impl(TransformType.Grad)89 @custom_function_call.py_impl(TransformType.Jvp)108 unwrapped_output = custom_function_call(276 @custom_function_call.py_impl(TransformType.Vmap)335 return custom_function_call(op, *operands)364 output = custom_function_call(vmapped_function, *unwrapped_operands)370 @custom_function_call.py_impl(TransformType.Functionalize)
15 from torch._functorch.autograd_function import custom_function_call585 return custom_function_call(cls, *args, **kwargs)