Searched refs:parameters_and_names (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/nn/layer/ |
| D | container.py | 55 _, param = next(cell.parameters_and_names())
204 for _, param in cell.parameters_and_names():
329 for _, param in cell.parameters_and_names():
354 for _, param in tmp_cell.parameters_and_names():
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_pipeline_split.py | 126 for _, param in model._train_network.parameters_and_names():
143 for _, param in model._train_network.parameters_and_names():
215 for _, param in model._train_network.parameters_and_names():
233 for _, param in model._train_network.parameters_and_names():
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | cell_wrapper.py | 104 self.pull_weight_indices = self._pull_weight_inputs(self.network.parameters_and_names())
108 self.push_weight_indices = self._push_weight_inputs(self.network.parameters_and_names())
|
| /third_party/mindspore/mindspore/nn/probability/transforms/ |
| D | transform_bnn.py | 135 for value, param in self.backbone.parameters_and_names():
195 for value, param in self.backbone.parameters_and_names():
|
| /third_party/mindspore/mindspore/nn/ |
| D | cell.py | 925 for name, param in self.parameters_and_names(expand=recurse):
977 for _, param in self.parameters_and_names(expand=expand):
985 for value, param in self.parameters_and_names():
991 def parameters_and_names(self, name_prefix='', expand=True): member in Cell
1326 params = self.parameters_and_names()
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | cell_wrapper.py | 212 …f.get_weights_by_key_input_3 = self._get_weights_by_key_inputs(self.network.parameters_and_names())
216 …over_weights_by_key_input_3 = self._over_weights_by_key_inputs(self.network.parameters_and_names())
|
| /third_party/mindspore/mindspore/train/ |
| D | serialization.py | 242 for _, param in save_obj.parameters_and_names():
548 for _, param in net.parameters_and_names():
591 for _, param in net.parameters_and_names():
1368 for _, param in network.parameters_and_names():
|
| /third_party/mindspore/tests/ut/python/pynative_mode/nn/ |
| D | test_cell.py | 132 for m in n.parameters_and_names():
|
| /third_party/mindspore/tests/ut/python/nn/ |
| D | test_cell.py | 124 for m in n.parameters_and_names():
|
| /third_party/mindspore/mindspore/compression/quant/ |
| D | quant_utils.py | 326 for name, param in quant_model.parameters_and_names():
|