Searched refs:grad_mindspore_impl (Results 1 – 4 of 4) sorted by relevance
/third_party/mindspore/tests/st/auto_monad/ |
D | test_auto_monad_mindtester.py | 311 def grad_mindspore_impl(self, params, grad_ys): member in SideEffectMemoryCellAddnNet 326 net.grad_mindspore_impl(inputs, grad_ys) 343 def grad_mindspore_impl(self, params, grad_ys): member in SideEffectIOCellAddnNet 359 net.grad_mindspore_impl(inputs, grad_ys) 376 def grad_mindspore_impl(self, params, grad_ys): member in SideEffectReturnParameterNet 391 net.grad_mindspore_impl(inputs, grad_ys) 409 def grad_mindspore_impl(self, params, grad_ys): member in SideEffectAssignAddnReluReturnParNet 424 out1 = net.grad_mindspore_impl(inputs, grad_ys) 428 out2 = net.grad_mindspore_impl(inputs, grad_ys) 459 def grad_mindspore_impl(self, params, grad_ys): member in SideEffectPrintInHighOrdeAddnNet [all …]
|
D | test_auto_monad_gpu.py | 160 def grad_mindspore_impl(self, params1, params2, params3, grad_ys): member in SideEffectControlFlowAssignDependWhileNet 223 def grad_mindspore_impl(self, params, grad_ys): member in SideEffectTwoAssignTwoAddnDependencyNet
|
/third_party/mindspore/tests/st/pynative/ |
D | test_pynative_layernorm_input_and_argmaxwithvalue.py | 118 def grad_mindspore_impl(self): member in LayerNormFactory 139 graph_grad1, graph_grad2, graph_grad3 = self.grad_mindspore_impl() 142 pynative_grad1, pynative_grad2, pynative_grad3 = self.grad_mindspore_impl() 181 def grad_mindspore_impl(self): member in ArgMaxWithValueFactory 202 graph_grad = self.grad_mindspore_impl() 205 pynative_grad = self.grad_mindspore_impl()
|
/third_party/mindspore/tests/ut/python/parameter_feature/ |
D | test_var_grad.py | 338 def grad_mindspore_impl(self): member in test_grad_for_concat.ConcatFactory 348 self.grad_mindspore_impl()
|