mindspore/tests/st/gradient
i-robot de4b5b238e
!40686 [bug] fix grad aux in pynative mode
Merge pull request !40686 from chenzhuo/master_aux
2022-08-24 01:05:01 +00:00
..
test_custom_cell_bprop.py Support custom cell bprop when using Parameter in primal graph 2022-04-27 14:08:17 +08:00
test_function_jvp_graph.py add function jvp 2021-11-12 08:44:41 +08:00
test_function_jvp_pynative.py add function jvp 2021-11-12 08:44:41 +08:00
test_function_linearize_graph.py add linearize test case 2022-08-10 11:23:42 +08:00
test_function_vjp_graph.py add taylor higher order differentiation 2022-03-10 19:26:14 +08:00
test_function_vjp_pynative.py add taylor higher order differentiation 2022-03-10 19:26:14 +08:00
test_grad_graph.py support grad and value_and_grad with has_aux 2022-08-12 16:43:03 +08:00
test_grad_mutable.py Fix the lost mutable attr for ms_function inputs 2022-08-01 14:42:49 +08:00
test_grad_pynative.py fix aux of grad in pynative mode 2022-08-23 14:32:28 +08:00
test_jvp_graph.py Vjp and fix jvp 2021-08-27 18:26:57 +08:00
test_jvp_pynative.py Vjp and fix jvp 2021-08-27 18:26:57 +08:00
test_taylor_differentiation_graph.py support operator Tan in Taylor differentiation 2022-07-05 11:47:58 +08:00
test_taylor_differentiation_pynative.py add taylor higher order differentiation 2022-03-10 19:26:14 +08:00
test_vjp_graph.py Vjp and fix jvp 2021-08-27 18:26:57 +08:00
test_vjp_pynative.py Add st and ut for jvp vjp and grad 2021-08-28 17:34:23 +08:00