mindspore/tests/st/gradient
chenzhuo 5f39ebd788 support grad and value_and_grad with has_aux 2022-08-12 16:43:03 +08: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 for compute jvp 2022-07-29 15:04:00 +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 support grad and value_and_grad with has_aux 2022-08-12 16:43:03 +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