Home
last modified time | relevance | path

Searched defs:tensor_grad_scale (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/tests/ut/python/pipeline/parse/
Dtest_for_stmt.py99 def tensor_grad_scale(x, op): function
/third_party/mindspore/tests/ut/python/parallel/
Dtest_loss_scale.py41 def tensor_grad_scale(scale, grad): function
/third_party/mindspore/mindspore/boost/
Dboost_cell_wrapper.py59 def tensor_grad_scale(scale, grad): function
/third_party/mindspore/tests/ut/python/ops/
Dtest_multitype_ops.py143 def tensor_grad_scale(scale, grad): function
/third_party/mindspore/mindspore/nn/wrap/
Dloss_scale.py33 def tensor_grad_scale(scale, grad): function
/third_party/mindspore/tests/st/fl/albert/src/
Dcell_wrapper.py97 def tensor_grad_scale(scale, grad): function
/third_party/mindspore/tests/st/networks/models/bert/src/
Dbert_for_pre_training.py333 def tensor_grad_scale(scale, grad): function
/third_party/mindspore/mindspore/nn/optim/
Doptimizer.py719 def tensor_grad_scale(scale, grad): function