Searched defs:fusion_type (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/wrap/ |
| D | grad_reducer.py | 85 def _init_allreduce_operators_by_parameters(parameters, split_indices, group, fusion_type=1): argument
413 …def __init__(self, parameters, mean=None, degree=None, fusion_type=1, group=GlobalComm.WORLD_COMM_… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/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 | 675 auto fusion_type = parallel_info.fusion_info()->FusionType(); in SetFusionInfoAttrToNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | kernel_build_info.h | 140 std::string fusion_type() const { return fusion_type_; } in fusion_type() function
|
| D | kernel_build_info.cc | 351 void KernelBuildInfo::KernelBuildInfoBuilder::SetFusionType(const std::string &fusion_type) { in SetFusionType()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/graph_kernel/model/ |
| D | graph_parallel.py | 23 def __init__(self, fusion_type, bottleneck, gain, block_assign, type_info): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ir/ |
| D | param_info.h | 90 void set_comm_fusion(int32_t fusion_type) { fusion_type_ = fusion_type; } in set_comm_fusion()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/ |
| D | op_info_register.py | 466 def fusion_type(self, fusion_type): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/ |
| D | cell.py | 2301 def set_comm_fusion(self, fusion_type, recurse=True): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/graph_kernel/ |
| D | graph_kernel_json_generator.cc | 1090 std::string fusion_type; in GenParallelJson() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.cc | 888 int32_t fusion_type = param_info->comm_fusion(); in AddCommOpFusionType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | kernel_graph_mgr.cc | 272 auto fusion_type = kernel_info_value[kFusionType]; in LoadAnfSelectKernelBuildInfo() local
|