Lines Matching full:backward

62     def backward(ctx, *grads):  function
74             f"Trying to backward through {op} but no autograd "
84             "backward": staticmethod(backward),
112     orig_backward = cls.backward
149                 "NYI: calling supports_tensorlist autograd.Function.backward directly. "
174         # Assume that any Nones in the backward are Tensors.
175         # If the forward has an arg that is [1, 2, 3], the backward should
177         # If the forward has an arg that is [tensor, tensor], the backward
184                 f"Expected the return from backward to be of the same structure "
185                 f"as the inputs. Got: {grad_inputs_spec} (return from backward), "
203     cls.backward = new_backward