Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dspliter.cc69 if (nodes_outputs_[node].size() > kDefaultBatch) { in RecordGraphInfo()
107 if (nodes_outputs_[curr_cnode].size() > kDefaultBatch) { in RecordGraphInfo()
Dmulti_conv_info.cc114 if (it == graph_node_outputs.end() || it->second.size() > kDefaultBatch) { in GetMultiConvNodes()
136 if (it->second.size() > kDefaultBatch) { in GetMultiConvNodes()
216 for (size_t j = kDefaultBatch + 1; j < ori_conv_cnode->size(); j++) { in AdJustInputs()
Dsplit_strategy.h50 constexpr auto kDefaultBatch = 1; variable
Dparallel_pass.cc114 if (output_info_list.size() > kDefaultBatch) { in Run()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dmulti_conv_split_pass.cc81 if (output_info_list.size() > kDefaultBatch) { in Process()
/third_party/mindspore/mindspore/lite/tools/converter/
Danf_transform.cc234 if (batch > opt::kDefaultBatch) { in RunParallelPass()
259 for (int match_number = max_match_number; match_number > opt::kDefaultBatch; --match_number) { in RunParallelPass()