David González Martínez 724e408736 add support for retain_graph in backward (#6145) ...

* add support for retain_graph in backward

* fix: dont accumulate grad on non-leaf tensors

* fix order

* fix: do not delete grad on leafs

* fix linter

* fix: can't exactly match torch behaviour internally

* allow numerical room for test

* refactor

2024-08-18 16:08:31 -07:00

29 KiB

Raw Blame History

View Raw