Searched refs:splits_ (Results 1 – 4 of 4) sorted by relevance
138 splits_ = strategys[0][kAxisN]; in CheckStrategy()145 splits_ = strategys[0][kAxisCIn]; in CheckStrategy()152 splits_ = strategys[1][kAxisCOut]; in CheckStrategy()159 splits_ = strategys[0][kAxisH]; in CheckStrategy()186 size_t dev_num = splits_.size(); in CheckSplitOutputs()212 int64_t dev_num = static_cast<int64_t>(splits_.size()); in AdJustConvPrim()213 int64_t total_ratio = std::accumulate(splits_.begin(), splits_.end(), 0); in AdJustConvPrim()238 auto curr_channel = in_channel * splits_.at(output_conv_index) / total_ratio; in AdJustConvPrim()247 auto curr_channel = out_channel * splits_.at(output_conv_index) / total_ratio; in AdJustConvPrim()256 auto curr_group = group * splits_.at(output_conv_index) / total_ratio; in AdJustConvPrim()[all …]
50 splits_ = strategys[0][kAxisN]; in CheckStrategy()56 splits_ = strategys[0][kAxisCIn]; in CheckStrategy()66 splits_ = strategys[1][kAxisCOut]; in CheckStrategy()72 splits_ = strategys[0][kAxisH]; in CheckStrategy()123 total_ratio = std::accumulate(splits_.begin(), splits_.end(), total_ratio); in CheckIfSplit()246 …re_split_cnode = CreateOutputsOfSplit(cnode_, 0, &feature_split_outputs, kAxisH, dev_num, splits_); in InferParallelCNodes()
49 std::vector<int64_t> splits_; variable
343 splits_.Set(255); in ByteMapBuilder()355 Bitmap256 splits_; member in re2::ByteMapBuilder387 if (0 <= lo && !splits_.Test(lo)) { in Merge()388 splits_.Set(lo); in Merge()389 int next = splits_.FindNextSetBit(lo+1); in Merge()392 if (!splits_.Test(hi)) { in Merge()393 splits_.Set(hi); in Merge()394 int next = splits_.FindNextSetBit(hi+1); in Merge()400 int next = splits_.FindNextSetBit(c); in Merge()417 int next = splits_.FindNextSetBit(c); in Build()