Home
last modified time | relevance | path

Searched refs:GradOfAllInputs (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/tests/ut/python/pynative_mode/
Dtest_multi_grad.py65 class GradOfAllInputs(_Grad): class
102 grad_mul = GradOfAllInputs(mulnet)
103 grad_add = GradOfAllInputs(addnet)
135 grad_mul = GradOfAllInputs(mulnet)
170 grad_mul = GradOfAllInputs(addnet)
171 grad_add = GradOfAllInputs(mulnet)
208 grad_mul = GradOfAllInputs(addnet)
Dtest_graph_param_cases.py67 class GradOfAllInputs(_Grad): class
135 net_me = GradOfAllInputs(net, real_inputs_count=3)
/third_party/mindspore/tests/st/pynative/
Dtest_pynative_hook_grad.py91 class GradOfAllInputs(_Grad): class
236 grad_net = GradOfAllInputs(ms_net)
248 grad_net = GradOfAllInputs(ms_net)
271 grad_net = GradOfAllInputs(ms_net)
297 grad_net = GradOfAllInputs(ms_net)
315 grad_net = GradOfAllInputs(ms_net)
333 grad_net = GradOfAllInputs(ms_net)
352 grad_net = GradOfAllInputs(ms_net)
374 grad_net = GradOfAllInputs(ms_net)
402 grad_net = GradOfAllInputs(ms_net)
Dtest_graph_param_transform.py65 class GradOfAllInputs(_Grad): class
/third_party/mindspore/tests/st/auto_monad/
Dtest_auto_monad_mindtester.py66 class GradOfAllInputs(_Grad): class
199 grad_net = GradOfAllInputs(net, sens_param=False)
223 self.grad = GradOfAllInputs(self.f, sens_param=False)
248 grad_net = GradOfAllInputs(net, sens_param=False)
Dtest_auto_monad_gpu.py60 class GradOfAllInputs(_Grad): class