Searched defs:save_gradient (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/nn/layer/ | ||
| D | thor_layer.py | 144 def save_gradient(self, dout): member in DenseThor |
| /third_party/mindspore/tests/ut/python/pynative_mode/ | ||
| D | test_insert_grad_of.py | 134 def save_gradient(self, dout): member in test_cell_assign.Mul |
| /third_party/mindspore/tests/ut/python/pipeline/infer/ | ||
| D | test_auto_monad.py | 118 def save_gradient(self, dout): member in test_insert_gradient_of.InsertGradientNet |
| /third_party/mindspore/tests/ut/python/pipeline/parse/ | ||
| D | test_parse.py | 273 def save_gradient(self, dout): member in AssignWhenInsertGrad |