Searched defs:requires_grad (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/graph_utils/python_pass/ |
| D | python_pass_register.py | 41 def __init__(self, requires_grad=True, run_only_once=False): argument
90 def register_pass(requires_grad=True, run_only_once=False): argument
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | param_info.h | 42 bool requires_grad() const { return requires_grad_; } in requires_grad() function
43 void set_requires_grad(bool requires_grad) { requires_grad_ = requires_grad; } in set_requires_grad()
|
| /third_party/mindspore/mindspore/common/ |
| D | parameter.py | 133 …def __init__(self, default_input, name=None, requires_grad=True, layerwise_parallel=False, paralle… argument
460 def requires_grad(self): member in Parameter
465 def requires_grad(self, value=True): member in Parameter
|
| /third_party/mindspore/mindspore/graph_utils/ |
| D | graph_pattern.py | 181 def __init__(self, para_name, default_tensor, requires_grad=False, layerwise_parallel=False): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | py_pass_manager.cc | 52 bool requires_grad, bool run_only_once) { in Register()
|
| D | pattern.h | 218 …explicit NewParameter(string para_name, tensor::TensorPtr default_tensor, bool requires_grad, bool… in NewParameter()
231 bool requires_grad() { return requires_grad_; } in requires_grad() function
|
| D | py_pass.cc | 200 bool requires_grad, bool layerwise_parallel) { in ReflectParamBackToPython()
|
| /third_party/mindspore/mindspore/nn/ |
| D | cell.py | 1215 def set_grad(self, requires_grad=True): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 481 auto requires_grad = param_info->requires_grad(); in ParameterColoring() local
|