| /third_party/mindspore/tests/ut/python/pynative_mode/ |
| D | test_multi_grad.py | 98 mulnet.set_grad()
99 addnet.set_grad()
131 mulnet.set_grad()
132 addnet.set_grad()
166 mulnet.set_grad()
167 addnet.set_grad()
204 mulnet.set_grad()
205 addnet.set_grad()
|
| D | test_kw_and_kwarg.py | 82 kw_net.set_grad(True)
127 net.set_grad(True)
|
| /third_party/mindspore/tests/st/pynative/ |
| D | test_pynative_hook_grad.py | 231 ms_net.set_grad()
244 ms_net.set_grad()
265 ms_net.set_grad()
293 ms_net.set_grad()
310 ms_net.set_grad()
329 ms_net.set_grad()
347 ms_net.set_grad()
369 ms_net.set_grad()
396 ms_net.set_grad()
|
| D | test_pynative_resnet50_gpu.py | 394 net_with_criterion.set_grad()
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | container.py | 215 def set_grad(self, flag=True): member in SequentialCell
218 cell.set_grad(flag)
389 def set_grad(self, flag=True): member in CellList
392 cell.set_grad(flag)
|
| /third_party/mindspore/mindspore/explainer/explanation/_attribution/ |
| D | attribution.py | 38 self._network.set_grad(False)
|
| /third_party/mindspore/mindspore/explainer/explanation/_attribution/_backprop/ |
| D | gradient.py | 82 self._backward_model.set_grad(False)
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | base.py | 355 fn.set_grad()
357 fn.set_grad(False)
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | cell_wrapper.py | 196 self.network.set_grad()
277 self.network.set_grad()
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/ |
| D | network_define.py | 134 self.network.set_grad()
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | cell_wrapper.py | 81 self.network.set_grad()
|
| /third_party/mindspore/mindspore/nn/wrap/ |
| D | cell_wrapper.py | 264 self.network.set_grad()
338 self.network.set_grad()
|
| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_tensor_copy_slices.py | 84 net.set_grad()
|
| /third_party/mindspore/mindspore/lite/examples/export_models/models/ |
| D | tinybert_train_export.py | 117 self.network.set_grad()
|
| /third_party/mindspore/tests/st/networks/ |
| D | test_gradient_accumulation.py | 78 self.network.set_grad()
|
| /third_party/mindspore/tests/st/pynative/loss_scale/ |
| D | test_loss_scale.py | 248 net_with_loss.set_grad()
|
| /third_party/mindspore/tests/ut/python/pipeline/parse/ |
| D | test_parse.py | 259 net.set_grad()
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_dense_op.py | 314 net.set_grad()
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | yolo.py | 414 self.network.set_grad()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | anf.h | 207 void set_grad(const bool &need_grad) { need_grad_ = need_grad; } in set_grad() function
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_auto_monad_mindtester.py | 222 self.f.set_grad()
|
| /third_party/mindspore/mindspore/nn/ |
| D | cell.py | 1215 def set_grad(self, requires_grad=True): member in Cell
|