| /third_party/mindspore/mindspore/nn/wrap/ |
| D | grad_reducer.py | 52 def _init_allreduce_operators_by_parameters(parameters, split_indices, group, fusion_type=1): argument
366 …def __init__(self, parameters, mean=True, degree=None, fusion_type=1, group=GlobalComm.WORLD_COMM_… argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_build_info.h | 89 FusionType fusion_type() const { return fusion_type_; } in fusion_type() function
|
| D | kernel_build_info.cc | 176 void KernelBuildInfo::KernelBuildInfoBuilder::SetFusionType(FusionType fusion_type) { in SetFusionType()
|
| D | common_utils.cc | 417 auto fusion_type = GetFusionTypeByName(fusion_name); in SetKernelBuildInfo() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | parallel_cost_model.cc | 82 FusionInfoPtr ParallelCostModel::ProcessFusionInfo(const py::object &fusion_type, const py::object … in ProcessFusionInfo() argument
|
| D | parallel_fusion.cc | 666 auto fusion_type = parallel_info.fusion_info()->FusionType(); in SetFusionInfoAttrToNode() local
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | param_info.h | 81 void set_comm_fusion(int32_t fusion_type) { fusion_type_ = fusion_type; } in set_comm_fusion()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/ |
| D | opinfo.h | 129 std::string fusion_type() const { return fusion_type_; } in fusion_type() function
147 void set_fusion_type(const std::string &fusion_type) { fusion_type_ = fusion_type; } in set_fusion_type()
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/model/ |
| D | graph_parallel.py | 22 def __init__(self, fusion_type, bottleneck, gain, block_assign, type_info): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.cc | 160 auto fusion_type = kernel::GetFusionTypeByName(fusion_name); in PreTaskFinishProcess() local
|
| /third_party/mindspore/mindspore/ops/ |
| D | op_info_register.py | 188 def fusion_type(self, fusion_type): member in RegOp
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_select.cc | 297 auto fusion_type = kernel::GetFusionTypeByName(fusion_name); in SetTbeBuildCommonInfo() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_generator.cc | 738 std::string fusion_type; in GenParallelJson() local
|
| /third_party/mindspore/mindspore/nn/ |
| D | cell.py | 1330 def set_comm_fusion(self, fusion_type, recurse=True): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.cc | 360 int32_t fusion_type = param_info->comm_fusion(); in AddCommOpFusionType() local
|