Searched refs:all_reduce_fusion_split_sizes_ (Results 1 – 2 of 2) sorted by relevance
68 all_reduce_fusion_split_sizes_.clear(); in Reset()176 all_reduce_fusion_split_sizes_[group + TypeIdLabel(kNumberTypeFloat)] = sizes; in SetAllReduceFusionSplitSizes()177 all_reduce_fusion_split_sizes_[group + TypeIdLabel(kNumberTypeFloat16)] = sizes; in SetAllReduceFusionSplitSizes()178 all_reduce_fusion_split_sizes_[group + TypeIdLabel(kNumberTypeFloat32)] = sizes; in SetAllReduceFusionSplitSizes()180 all_reduce_fusion_split_sizes_[group] = sizes; in SetAllReduceFusionSplitSizes()184 auto iter = all_reduce_fusion_split_sizes_.find(group); in GetAllReduceFusionSplitSizes()185 if (iter != all_reduce_fusion_split_sizes_.end()) { in GetAllReduceFusionSplitSizes()
161 std::map<std::string, std::vector<uint32_t>> all_reduce_fusion_split_sizes_; variable