Home
last modified time | relevance | path

Searched defs:grad_out (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/tests/st/hook/
Dtest_hook_function.py26 def var_hook_function(grad_out): argument
/third_party/mindspore/tests/ut/python/pynative_mode/
Dtest_hook.py62 def var_hook_function(grad_out): argument
/third_party/mindspore/tests/st/pynative/
Dtest_pynative_hook.py69 def custom_hook_function_print_and_save_grad(grad_out): argument
/third_party/mindspore/mindspore/ccsrc/pybind_api/ir/
Dprimitive_py.cc150 void PrimitivePy::CheckHookConsistency(const py::object &grad_out, const py::object &expected_grad_… in CheckHookConsistency() argument
/third_party/mindspore/tests/ut/python/optimizer/
Dtest_auto_grad.py590 def var_hook_function(grad_out): argument