Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/python/pynative_mode/
Dtest_multi_grad.py55 class GradOfFirstInput(_Grad): class
136 grad_add = GradOfFirstInput(mulnet)
209 grad_add = GradOfFirstInput(mulnet)
Dtest_graph_param_cases.py57 class GradOfFirstInput(_Grad): class
/third_party/mindspore/tests/st/high_grad/
Dtest_highgrad_train.py53 class GradOfFirstInput(_Grad): class
104 grad_net = GradOfFirstInput(net, sens_param=False)
/third_party/mindspore/tests/st/auto_monad/
Dtest_auto_monad_mindtester.py56 class GradOfFirstInput(_Grad): class
579 grad_net = GradOfFirstInput(self)
636 grad_net = HighGrad(net, [GradOfFirstInput, GradOfFirstInput])
649 net, [GradOfFirstInput, GradOfFirstInput, GradOfFirstInput])
/third_party/mindspore/tests/st/pynative/
Dtest_graph_param_transform.py55 class GradOfFirstInput(_Grad): class
196 net_me = GradOfFirstInput(net, real_inputs_count=1)
Dtest_pynative_layernorm_input_and_argmaxwithvalue.py156 class GradOfFirstInput(_Grad): class
188 grad_net = GradOfFirstInput(net, real_inputs_count=1)