Searched refs:comm_input_info_map (Results 1 – 1 of 1) sorted by relevance
294 std::map<AnfNodePtr, std::vector<size_t>> comm_input_info_map; in GetCommunicationOpInputInfo() local318 auto iter = comm_input_info_map.find(kernel_input.first); in GetCommunicationOpInputInfo()319 if (iter != comm_input_info_map.end()) { in GetCommunicationOpInputInfo()323 comm_input_info_map[kernel_input.first] = indexes; in GetCommunicationOpInputInfo()330 for (auto &info : comm_input_info_map) { in GetCommunicationOpInputInfo()335 return comm_input_info_map; in GetCommunicationOpInputInfo()356 …std::map<AnfNodePtr, std::vector<size_t>> comm_input_info_map = GetCommunicationOpInputInfo(kernel… in TbeClearZeroNodeFusion() local361 if (comm_input_info_map.find(anf_node) != comm_input_info_map.end()) { in TbeClearZeroNodeFusion()362 auto indexes = comm_input_info_map[anf_node]; in TbeClearZeroNodeFusion()426 …std::map<AnfNodePtr, std::vector<size_t>> comm_input_info_map = GetCommunicationOpInputInfo(kernel… in KernelBuildPreprocess() local[all …]