Home
last modified time | relevance | path

Searched refs:split_flag_list_ (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Darithmetic_info.cc144 split_flag_list_[0] = false; in ReComputeBatchSplitFlagList()
145 split_flag_list_[1] = false; in ReComputeBatchSplitFlagList()
148 (expend_a_shape.at(0) != 1) ? (split_flag_list_[0] = true) : (split_flag_list_[0] = false); in ReComputeBatchSplitFlagList()
149 (expend_b_shape.at(0) != 1) ? (split_flag_list_[1] = true) : (split_flag_list_[1] = false); in ReComputeBatchSplitFlagList()
Dbias_add_info.cc54 split_flag_list_[0] = true; in ReComputeBatchSplitFlagList()
55 split_flag_list_[1] = false; in ReComputeBatchSplitFlagList()
Dgathernd_info.cc115 split_flag_list_[0] = false; in ReComputeBatchSplitFlagList()
116 split_flag_list_[1] = true; in ReComputeBatchSplitFlagList()
Dbatch_parallel_info.cc157 if (split_flag_list_[i]) { in GenerateOpStrategies()
170 split_flag_list_[i] = true; in ReComputeBatchSplitFlagList()
Dtile_info.cc186 split_flag_list_ = {true}; in GenerateBatchStrategies()
187 return GenerateBatchStrategiesBySplitFlag(multiples_shape, split_flag_list_); in GenerateBatchStrategies()
Dslice_info.cc153 split_flag_list_ = {true}; in GenerateBatchStrategies()
154 return GenerateBatchStrategiesBySplitFlag(inputs_shape_, split_flag_list_); in GenerateBatchStrategies()
Dstrided_slice_info.cc206 split_flag_list_ = {true}; in GenerateBatchStrategies()
207 return GenerateBatchStrategiesBySplitFlag(inputs_shape_, split_flag_list_); in GenerateBatchStrategies()
Dselect_info.cc96 split_flag_list_[i] = true; in ReComputeBatchSplitFlagList()
Dloss_info.cc130 split_flag_list_[i] = true; in ReComputeBatchSplitFlagList()
Dvirtual_dataset_info.cc145 split_flag_list_[i] = true; in ReComputeBatchSplitFlagList()
Dpack_info.cc124 split_flag_list_[i] = true; in ReComputeBatchSplitFlagList()
Dscatter_update_info.cc136 split_flag_list_[i] = false; // the first dimension can not be split in ReComputeBatchSplitFlagList()
Dconcat_info.cc142 split_flag_list_[i] = true; in ReComputeBatchSplitFlagList()
Dgatherd_info.cc188 split_flag_list_[i] = true; in ReComputeBatchSplitFlagList()
Doperator_info.cc993 split_flag_list_[0] = true; in ReComputeBatchSplitFlagList()
998 split_flag_list_.clear(); in ComputeBatchSplitFlagList()
1000 split_flag_list_.push_back(false); in ComputeBatchSplitFlagList()
1036 return GenerateBatchStrategiesBySplitFlag(inputs_shape_, split_flag_list_); in GenerateBatchStrategies()
Doperator_info.h285 std::vector<bool> split_flag_list_; variable
Dconv2d_info.cc826 split_flag_list_[0] = true; in ReComputeBatchSplitFlagList()
827 split_flag_list_[1] = false; in ReComputeBatchSplitFlagList()