Home
last modified time | relevance | path

Searched refs:combined_partitions (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dmatmul_info.cc427 Dimensions combined_partitions; in GenerateStrategies() local
437 …std::function<void(uint64_t, size_t)> recursive = [&stage_id, &dev_num, &combined_partitions, &com… in GenerateStrategies()
444 …if (this->PrepareStrategy(stage_id, dev_num, combined_partitions, input0_shape_size, input1_shape_… in GenerateStrategies()
457 combined_partitions.push_back(i); in GenerateStrategies()
459 combined_partitions.pop_back(); in GenerateStrategies()
477 … mindspore::parallel::Dimensions combined_partitions, size_t input0_shape_size, in PrepareStrategy() argument
480 …std::accumulate(combined_partitions.begin(), combined_partitions.end(), 1, std::multiplies<int64_t… in PrepareStrategy()
494 input0_partitions.push_back(combined_partitions[i]); in PrepareStrategy()
497 input1_partitions.push_back(combined_partitions[combined_partitions.size() - 2]); in PrepareStrategy()
498 input1_partitions.push_back(combined_partitions[combined_partitions.size() - 1]); in PrepareStrategy()
[all …]
Dtensordot_info.h52 …Status PrepareStrategy(int32_t stage_id, size_t dev_num, Dimensions combined_partitions, size_t in…
Dmatmul_info.h47 …Status PrepareStrategy(int64_t stage_id, size_t dev_num, Dimensions combined_partitions, size_t in…
Doperator_info.cc1264 Shape combined_inputs_shape, combined_splittable_inputs, combined_partitions; in GenerateStrategiesForIndependentInputs() local
1271 … &combined_splittable_inputs, &combined_partitions, &recursive, in GenerateStrategiesForIndependentInputs()
1280 tmp_partition.push_back(combined_partitions[global_index]); in GenerateStrategiesForIndependentInputs()
1293 combined_partitions.push_back(MIN_SLICE_NUM); in GenerateStrategiesForIndependentInputs()
1295 combined_partitions.pop_back(); in GenerateStrategiesForIndependentInputs()
1299 combined_partitions.push_back(i); in GenerateStrategiesForIndependentInputs()
1301 combined_partitions.pop_back(); in GenerateStrategiesForIndependentInputs()