diff --git a/mobile/src/io/opencl_interface.h b/mobile/src/io/opencl_interface.h index d7877e8afd0e1cf40b732cfa15719735d9714a8c..f1039f1373df6b65a5fc7f4e01279badfaa40307 100644 --- a/mobile/src/io/opencl_interface.h +++ b/mobile/src/io/opencl_interface.h @@ -22,6 +22,6 @@ cl_context getContext(); cl_command_queue getClCommandQueue(); bool isInitSuccess(); -#endif - } // namespace paddle_mobile + +#endif diff --git a/mobile/src/operators/op_param.h b/mobile/src/operators/op_param.h index 161d8105697f811fe83774d377103466cc4d2f17..d31a5e6c21a9c874f2b2f6650f807032ddc665e2 100644 --- a/mobile/src/operators/op_param.h +++ b/mobile/src/operators/op_param.h @@ -687,7 +687,7 @@ class ConcatParam : public OpParam { inputs_ = InputMultiFrom(inputs, *scope); out_ = OutFrom(outputs, *scope); axis_ = GetAttr("axis", attrs); - original_output_dims_size_ = inputs_[0]->dims().size(); + original_output_dims_size_ = out_->dims().size(); } vector Inputs() const { return inputs_; }