Searched refs:PrepareStrategy (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | tensordot_info.h | 52 …Status PrepareStrategy(int32_t stage_id, size_t dev_num, Dimensions combined_partitions, size_t in…
|
| D | matmul_info.h | 47 …Status PrepareStrategy(int64_t stage_id, size_t dev_num, Dimensions combined_partitions, size_t in…
|
| D | matmul_info.cc | 444 …if (this->PrepareStrategy(stage_id, dev_num, combined_partitions, input0_shape_size, input1_shape_… in GenerateStrategies()
476 Status MatMulBase::PrepareStrategy(int64_t stage_id, size_t dev_num, in PrepareStrategy() function in mindspore::parallel::MatMulBase
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_generate_strategy.h | 64 Strategys PrepareStrategy(const std::shared_ptr<Graph> &graph, const std::vector<std::shared_ptr<Op…
|
| D | rec_generate_strategy.cc | 580 Strategys PrepareStrategy(const std::shared_ptr<Graph> &graph, const std::vector<std::shared_ptr<Op… in PrepareStrategy() function
617 strategies = PrepareStrategy(graph, ops, iter_graph, iter_ops); in GeneratePartitionedOperatorStrategy()
|