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

fix yolov3 concat op test=develop (#1901)

上级 111c0b6d
......@@ -22,6 +22,6 @@ cl_context getContext();
cl_command_queue getClCommandQueue();
bool isInitSuccess();
#endif
} // namespace paddle_mobile
#endif
......@@ -687,7 +687,7 @@ class ConcatParam : public OpParam {
inputs_ = InputMultiFrom<GType>(inputs, *scope);
out_ = OutFrom<GType>(outputs, *scope);
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_; }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册