Searched refs:idx_list (Results 1 – 3 of 3) sorted by relevance
44 def _set_fusion_strategy_by_idx(idx_list, group="hccl_world_group"): argument82 if isinstance(idx_list, list):83 idx_len = len(idx_list)89 for idx in idx_list:96 c_array_idx_list = _c_array(ctypes.c_uint, idx_list)97 c_idx_num = ctypes.c_uint(len(idx_list))
144 bool Map::CheckNoneByIdxList(std::vector<size_t> idx_list) const { in CheckNoneByIdxList()145 for (auto &value : idx_list) { in CheckNoneByIdxList()153 Map Map::SqueezeMapByIdxList(std::vector<size_t> idx_list) const { in SqueezeMapByIdxList()156 auto it = std::find(idx_list.begin(), idx_list.end(), i); in SqueezeMapByIdxList()157 if (it == idx_list.end()) { in SqueezeMapByIdxList()
43 bool CheckNoneByIdxList(std::vector<size_t> idx_list) const;44 Map SqueezeMapByIdxList(std::vector<size_t> idx_list) const;