未验证 提交 9b343653 编写于 作者: Y Yanzhan Yang 提交者: GitHub

fix yolov3 concat op test=develop (#1901)

上级 111c0b6d
...@@ -22,6 +22,6 @@ cl_context getContext(); ...@@ -22,6 +22,6 @@ cl_context getContext();
cl_command_queue getClCommandQueue(); cl_command_queue getClCommandQueue();
bool isInitSuccess(); bool isInitSuccess();
#endif
} // namespace paddle_mobile } // namespace paddle_mobile
#endif
...@@ -687,7 +687,7 @@ class ConcatParam : public OpParam { ...@@ -687,7 +687,7 @@ class ConcatParam : public OpParam {
inputs_ = InputMultiFrom<GType>(inputs, *scope); inputs_ = InputMultiFrom<GType>(inputs, *scope);
out_ = OutFrom<GType>(outputs, *scope); out_ = OutFrom<GType>(outputs, *scope);
axis_ = GetAttr<int>("axis", attrs); axis_ = GetAttr<int>("axis", attrs);
original_output_dims_size_ = inputs_[0]->dims().size(); original_output_dims_size_ = out_->dims().size();
} }
vector<GType *> Inputs() const { return inputs_; } vector<GType *> Inputs() const { return inputs_; }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册