提交 bb56b583 编写于 作者: W wanghaoshuang

Fix warning log in switch order layer

上级 4de84fb3
...@@ -32,10 +32,10 @@ bool SwitchOrderLayer::init(const LayerMap& layerMap, ...@@ -32,10 +32,10 @@ bool SwitchOrderLayer::init(const LayerMap& layerMap,
outDims_ = TensorShape(4); outDims_ = TensorShape(4);
auto& reshape_conf = config_.reshape_conf(); auto& reshape_conf = config_.reshape_conf();
for (size_t i = 0; i < reshape_conf.heightaxis_size(); i++) { for (int i = 0; i < reshape_conf.heightaxis_size(); i++) {
heightAxis_.push_back(reshape_conf.heightaxis(i)); heightAxis_.push_back(reshape_conf.heightaxis(i));
} }
for (size_t i = 0; i < reshape_conf.widthaxis_size(); i++) { for (int i = 0; i < reshape_conf.widthaxis_size(); i++) {
widthAxis_.push_back(reshape_conf.widthaxis(i)); widthAxis_.push_back(reshape_conf.widthaxis(i));
} }
createFunction(nchw2nhwc_, "NCHW2NHWC", FuncConfig()); createFunction(nchw2nhwc_, "NCHW2NHWC", FuncConfig());
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册