Home
last modified time | relevance | path

Searched refs:inputs_device_format (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dconcat_outputs_for_all_gather.cc53 std::vector<std::string> inputs_device_format; in GenerateKernelBuildInfo() local
61 inputs_device_format.emplace_back(std::get<2>(allgather_output_info)[input_index]); in GenerateKernelBuildInfo()
65 auto cmp_format = inputs_device_format.begin(); in GenerateKernelBuildInfo()
66 auto format_iter = std::find_if(inputs_device_format.begin(), inputs_device_format.end(), in GenerateKernelBuildInfo()
68 if (format_iter != inputs_device_format.end()) { in GenerateKernelBuildInfo()
82 builder.SetInputsFormat(inputs_device_format); in GenerateKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dconcat_outputs_for_all_gather.cc53 std::vector<std::string> inputs_device_format; in GenerateKernelBuildInfo() local
61 (void)inputs_device_format.emplace_back(std::get<kIndex2>(allgather_output_info)[input_index]); in GenerateKernelBuildInfo()
65 auto cmp_format = inputs_device_format.begin(); in GenerateKernelBuildInfo()
66 auto format_iter = std::find_if(inputs_device_format.begin(), inputs_device_format.end(), in GenerateKernelBuildInfo()
68 if (format_iter != inputs_device_format.end()) { in GenerateKernelBuildInfo()
84 builder.SetInputsFormat(inputs_device_format); in GenerateKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dreplace_node_by_proxy.cc27 std::vector<std::string> inputs_device_format; in GenerateKernelBuildInfo() local
35 inputs_device_format.push_back(AnfAlgo::GetInputFormat(cnode, input_index)); in GenerateKernelBuildInfo()
48 builder.SetInputsFormat(inputs_device_format); in GenerateKernelBuildInfo()
Dcommunication_op_fusion.cc42 std::vector<std::string> inputs_device_format; in GenerateKernelBuildInfo() local
61 inputs_device_format.push_back(AnfAlgo::GetInputFormat(cnode, input_index)); in GenerateKernelBuildInfo()
80 builder.SetInputsFormat(inputs_device_format); in GenerateKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/ps/
Dutil.cc206 std::vector<std::string> inputs_device_format; in GenerateKernelBuildInfo() local
217 (void)inputs_device_format.emplace_back(kOpFormat_DEFAULT); in GenerateKernelBuildInfo()
227 builder.SetInputsFormat(inputs_device_format); in GenerateKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc965 std::vector<std::string> inputs_device_format; in GenerateKernelBuildInfo() local
976 (void)inputs_device_format.emplace_back(kOpFormat_DEFAULT); in GenerateKernelBuildInfo()
986 builder.SetInputsFormat(inputs_device_format); in GenerateKernelBuildInfo()