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

!24030 Fix warning 

Merge pull request !24030 from LiangZhibo/warning
This commit is contained in:
i-robot 2021-09-27 07:34:13 +00:00 committed by Gitee
parent a9eb75ac42 59a8b0cdfa
commit b814ff1b70
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/core/abstract/prim_nn.cc
View File

@ -152,7 +152,7 @@ AbstractBasePtr InferImplBatchNorm(const AnalysisEnginePtr &, const PrimitivePtr
MS_EXCEPTION_IF_NULL(data_format_ptr);
int64_t data_format = GetAndCheckFormat(data_format_ptr);
int64_t c_axis = 1;
size_t c_axis = 1;
if (data_format == Format::NHWC) {
c_axis = 3;
}