Lines Matching full:backward
26 def backward(ctx, grad_output): member in CustomFunc1
41 def backward(ctx, grad_output): member in CustomFunc3
89 # Note that forward, setup_context, and backward are @staticmethods
106 def backward(ctx, grad_output): member in LinearFunction
131 def backward(ctx, grad_out1, grad_out2): member in MaterializingGradFunction
146 def backward(ctx, grad_output): member in CustomFuncBwdPrintGraphBreak
162 def backward(ctx, grad_output): member in CustomFuncStrideBwd
180 def backward(ctx, grad_output): member in CustomFuncSaveForBwd
199 def backward(ctx, grad_output): member in ContextSaveAndMark
212 def backward(ctx, grad_output): member in ContextMarkAndSave
309 def backward(ctx, g): member in AutogradFunctionTests.test_enum_arg.Foo
343 def backward(ctx, grad_out): member in AutogradFunctionTests.test_allow_in_graph.AllowInGraphFunc
369 def backward(ctx, grad): member in AutogradFunctionTests.test_once_differentiable.ScaleGradient
389 def backward(cls, ctx, grad_output): member in AutogradFunctionTests.test_classmethod.Shake
429 def backward(ctx, grad1, grad2): member in AutogradFunctionTests.test_multi_output.Foo
455 def backward(ctx, grad): member in AutogradFunctionTests.test_amp_custom_fwd_bwd.MyMM
466 res.backward(grad)
479 def backward(ctx, grad_out): member in AutogradFunctionTests.test_set_materialize_grads_no_graph_break.MulY
488 result.sum().backward()
508 def backward(ctx, grad): member in AutogradFunctionTests.test_user_defined_object_as_input.Foo
521 result.sum().backward()
576 def backward(ctx, grad): member in AutogradFunctionTests.test_tensor_list_as_input.Foo
591 result.sum().backward()
612 def backward(ctx, grad): member in AutogradFunctionTests.test_multiple_different_non_tensor_inputs.Foo
633 result.sum().backward()
651 def backward(ctx, grad): member in AutogradFunctionTests.test_backward_returns_none_for_tensor_input.Foo
663 result.sum().backward()
677 def backward(ctx, grad_output): member in AutogradFunctionTests.test_function_with_bound_free_variable.LowerBound
713 def backward(ctx, grad_output): member in AutogradFunctionTests.test_smoke_from_test_autograd.Mult
727 def backward(ctx, grad_output): member in AutogradFunctionTests.test_smoke_from_test_autograd.Double
731 # this is equivalent, but uses the output of .forward() in .backward()
734 def backward(ctx, grad_output): member in AutogradFunctionTests.test_smoke_from_test_autograd.Double2
747 def backward(ctx, grad_a, grad_b): member in AutogradFunctionTests.test_smoke_from_test_autograd.Identity
756 def backward(ctx, gO): member in AutogradFunctionTests.test_smoke_from_test_autograd.MyFunc2
769 def backward(ctx, grad): member in AutogradFunctionTests.test_smoke_from_test_autograd.MyFn
780 def backward(ctx, grad): member in AutogradFunctionTests.test_smoke_from_test_autograd.MyAdder
791 def backward(ctx, grad_output): member in AutogradFunctionTests.test_smoke_from_test_autograd.InplaceMul
823 def backward(ctx, grad): member in AutogradFunctionTests.test_smoke_from_test_autograd.MyFn3
845 c.sum().backward()
921 def backward(ctx, g): member in AutogradFunctionTests.test_tensor_subclass_intermediary_input.foo_autograd_fn
944 y_ref.sum().backward()
948 y.sum().backward()
963 def backward(ctx, grad_out): member in AutogradFunctionTests.test_smuggle_symint_issue_111031.Foo
987 def backward(ctx, grad_output): member in AutogradFunctionTests.test_needs_input_grad.NeedsInputGradFunc
994 NeedsInputGradFunc.apply(x).sum().backward()
1010 def backward(ctx, grad_out): member in AutogradFunctionTests.test_repeated_save_for_backward_calls.Foo
1025 out_ref.sum().backward()
1028 out_test.sum().backward()
1046 def backward(ctx, grad_out): member in AutogradFunctionTests.test_smuggle_tensor_and_complex_structures.Foo
1074 def backward(ctx, grad1, grad2): member in AutogradFunctionTests.test_mark_non_differentiable.MyFunction
1092 res1.sum().backward()
1146 def backward(ctx, grad1, grad2, grad3): member in AutogradFunctionTests.test_mark_multi_output_non_differentiable.MyFunction
1168 res1.sum().backward()
1179 def backward(ctx, grad_out): member in AutogradFunctionTests.test_default_values.Foo
1200 def backward(ctx, grad_output): member in AutogradFunctionTests.test_tuple_arg.TupleArgFunc
1211 out.sum().backward()
1232 def backward(ctx, grad_output): member in AutogradFunctionTests.test_triton_kernel_basic.Add
1245 loss.backward()
1265 def backward(ctx, grad_output, old_x): member in AutogradFunctionTests.test_triton_kernel_multiple_out.Add
1280 loss.backward()