Searched refs:broadcast_params (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline_ge.cc | 178 const std::string &phase, const py::object &broadcast_params) { in AddDFGraph() argument
194 if (!broadcast_params.is_none()) { in AddDFGraph()
195 if (!py::isinstance<py::dict>(broadcast_params)) { in AddDFGraph()
199 py::dict broadcast = broadcast_params.cast<py::dict>(); in AddDFGraph()
242 const std::string &phase, const py::object &broadcast_params) { in BuildDFGraph() argument
254 if (!AddDFGraph(info, init_params, phase, broadcast_params)) { in BuildDFGraph()
|
| D | pipeline_ge.h | 44 const std::string &phase, const py::object &broadcast_params = {});
|
| D | pipeline.h | 94 const py::object &broadcast_params = {});
|
| D | pipeline.cc | 1097 const py::object &broadcast_params) { in BuildGraph() argument
1099 return BuildDFGraph(info_, init_params, phase, broadcast_params); in BuildGraph()
|
| /third_party/mindspore/mindspore/common/ |
| D | api.py | 330 broadcast_params = []
332 broadcast_params.append(Tensor(param.asnumpy()))
333 _broadcast_net = _BroadCastCell(broadcast_params)
|
| /third_party/mindspore/tests/st/fl/mobile/src/ |
| D | adam.py | 317 self.broadcast_params(optim_result)
419 self.broadcast_params(optim_result)
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | adam.py | 317 self.broadcast_params(optim_result)
419 self.broadcast_params(optim_result)
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | adam.py | 316 self.broadcast_params(optim_result)
418 self.broadcast_params(optim_result)
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | lamb.py | 334 optim_result = F.depend(optim_result, self.broadcast_params(optim_result))
|
| D | adam.py | 509 self.broadcast_params(optim_result)
|
| D | optimizer.py | 620 def broadcast_params(self, optim_result): member in Optimizer
|