Home
last modified time | relevance | path

Searched refs:comm_fusion (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/nn/wrap/
Dgrad_reducer.py60 comm_fusion = parameter.comm_fusion
62 last_comm_fusion = comm_fusion
65 if comm_fusion != last_comm_fusion:
67 last_comm_fusion = comm_fusion
69 op.add_prim_attr('fusion', comm_fusion)
/third_party/mindspore/mindspore/common/
Dparameter.py308 def comm_fusion(self): member in Parameter
319 return self.param_info.comm_fusion
321 @comm_fusion.setter
322 def comm_fusion(self, comm_fusion_): member in Parameter
326 self.param_info.comm_fusion = comm_fusion_
/third_party/mindspore/mindspore/ccsrc/pybind_api/ir/
Dparam_info_py.cc34 … .def_property("comm_fusion", &ParamInfo::comm_fusion, &ParamInfo::set_comm_fusion) in __anon2c0500310102()
/third_party/mindspore/mindspore/core/ir/
Dparam_info.h80 int32_t comm_fusion() const { return fusion_type_; } in comm_fusion() function
/third_party/mindspore/mindspore/boost/
Dbase.py186 parameters[i].comm_fusion = self._parameter_indices
/third_party/mindspore/mindspore/nn/
Dcell.py1344 param.comm_fusion = fusion_type
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.cc360 int32_t fusion_type = param_info->comm_fusion(); in AddCommOpFusionType()