Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix maxpool log

This commit is contained in:
Ziyan 2021-07-08 17:05:40 +08:00
parent a27c7b2436
commit 0f854d89d9
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/frontend/parallel/ops_info/maxpool_info.cc
View File

@ -166,7 +166,8 @@ Status MaxPoolInfo::InferTensorMap() {
Status MaxPoolInfo::SetCostUnderStrategy(const StrategyPtr &strategy) { return SetCostUnderStrategyBase(strategy); }
std::vector<StrategyPtr> MaxPoolInfo::GenerateOpStrategies(int64_t stage_id) {
Shape input0_split(inputs_shape_[0].size(), 1);
Shape input0_split(inputs_shape_[0].size(), 0);
input0_split[0] = 1;
Shapes splittable_inputs = {input0_split};
std::vector<StrategyPtr> sp_vector;