Home
last modified time | relevance | path

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

/third_party/mindspore/tests/st/ops/ascend/
Dtest_rnn_op.py27 class GradOfAllInputsAndParams(nn.Cell): class
29 super(GradOfAllInputsAndParams, self).__init__()
159 grad_net_inp = GradOfAllInputsAndParams(net, sens_param=False)
174 grad_net_inp_pynative = GradOfAllInputsAndParams(net_pynative, sens_param=False)
Dtest_gru_op.py27 class GradOfAllInputsAndParams(nn.Cell): class
29 super(GradOfAllInputsAndParams, self).__init__()
159 grad_net_inp = GradOfAllInputsAndParams(net, sens_param=False)
174 grad_net_inp_pynative = GradOfAllInputsAndParams(net_pynative, sens_param=False)
Dtest_lstm_op.py28 class GradOfAllInputsAndParams(nn.Cell): class
156 grad_net_inp = GradOfAllInputsAndParams(net, sens_param=False)
170 grad_net_inp_pynative = GradOfAllInputsAndParams(net_pynative, sens_param=False)
/third_party/mindspore/tests/st/auto_monad/
Dtest_auto_monad_mindtester.py76 class GradOfAllInputsAndParams(_Grad): class
312 grad_net = GradOfAllInputsAndParams(self)
344 grad_net = GradOfAllInputsAndParams(self)
377 grad_net = GradOfAllInputsAndParams(self)
410 grad_net = GradOfAllInputsAndParams(self)
460 grad_net = GradOfAllInputsAndParams(self)
506 grad_net = GradOfAllInputsAndParams(self)
538 grad_net = GradOfAllInputsAndParams(self)
672 grad_net = GradOfAllInputsAndParams(self)
Dtest_auto_monad_gpu.py70 class GradOfAllInputsAndParams(_Grad): class
161 grad_net = GradOfAllInputsAndParams(self)
224 grad_net = GradOfAllInputsAndParams(self)
/third_party/mindspore/tests/st/pynative/
Dtest_pynative_layernorm_input_and_argmaxwithvalue.py50 class GradOfAllInputsAndParams(_Grad): class
123 grad_net = GradOfAllInputsAndParams(net)