Searched refs:CtxWithSavedTensors (Results 1 – 2 of 2) sorted by relevance
439 wrapped_ctx = CtxWithSavedTensors(ctx, saved_tensors)460 wrapped_ctx = CtxWithSavedTensors(ctx, saved_tensors)584 class CtxWithSavedTensors(WrappedCtx): class
3157 wrapped = torch._functorch.autograd_function.CtxWithSavedTensors(3171 wrapped = torch._functorch.autograd_function.CtxWithSavedTensors(3184 CtxWithSavedTensors = torch._functorch.autograd_function.CtxWithSavedTensors3197 ctx_y = CtxWithSavedTensors(ctx, (y,))3203 wrapped = CtxWithSavedTensors(ctx_y, (z,))3230 wrapped = torch._functorch.autograd_function.CtxWithSavedTensors(3253 wrapped = torch._functorch.autograd_function.CtxWithSavedTensors(