Searched refs:grad_ys (Results 1 – 2 of 2) 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): argument
314 grad_out = grad_net(params, grad_ys)
324 grad_ys = Tensor([18.0], ms.float32)
326 net.grad_mindspore_impl(inputs, grad_ys)
343 def grad_mindspore_impl(self, params, grad_ys): argument
346 grad_out = grad_net(params, grad_ys)
357 grad_ys = Tensor([18.0], ms.float32)
359 net.grad_mindspore_impl(inputs, grad_ys)
376 def grad_mindspore_impl(self, params, grad_ys): argument
379 grad_out = grad_net(params, grad_ys)
[all …]
|
| D | test_auto_monad_gpu.py | 160 def grad_mindspore_impl(self, params1, params2, params3, grad_ys): argument
163 grad_out = grad_net(params1, params2, params3, grad_ys)
223 def grad_mindspore_impl(self, params, grad_ys): argument
226 grad_out = grad_net(params, grad_ys)
|