diff --git a/paddle/fluid/lite/arm/math/split.cc b/paddle/fluid/lite/arm/math/split.cc index 6dd6de6242e806947dfc630fd8f2a4dd03c89335..bf8d50590ff89c451347e33a289391b8d929e5b6 100644 --- a/paddle/fluid/lite/arm/math/split.cc +++ b/paddle/fluid/lite/arm/math/split.cc @@ -52,10 +52,10 @@ void split_cpy(const float* din, float* dout, int num) { } template <> -void split(const float* din, std::vector* dout, +void split(const float* din, const std::vector& dout, const int axis, const std::vector& in_strides) { int input_offset = 0; - for (auto out : *dout) { + for (auto out : dout) { auto out_dim = out->dims(); std::vector out_strides(out_dim.size()); out_strides[out_dim.size() - 1] = out_dim[out_dim.size() - 1]; diff --git a/paddle/fluid/lite/arm/math/split.h b/paddle/fluid/lite/arm/math/split.h index 9b5651d81ffa75362fcc39db82157c56548917c0..643214e174c3ede02f430ee4ded7cee097ba0afc 100644 --- a/paddle/fluid/lite/arm/math/split.h +++ b/paddle/fluid/lite/arm/math/split.h @@ -26,7 +26,7 @@ template void split_cpy(const T* din, T* dout, int num); template -void split(const T* din, std::vector* dout, const int axis, +void split(const T* din, const std::vector& dout, const int axis, const std::vector& in_strides); } // namespace math diff --git a/paddle/fluid/lite/kernels/arm/split_compute.cc b/paddle/fluid/lite/kernels/arm/split_compute.cc index 9da69894592e146c9191eb9da38d8d481cf287a7..3c2416bd6907199e6e83baf65c428b675462f271 100644 --- a/paddle/fluid/lite/kernels/arm/split_compute.cc +++ b/paddle/fluid/lite/kernels/arm/split_compute.cc @@ -24,7 +24,7 @@ namespace arm { void SplitCompute::Run() { auto& param = Param(); const float* din = param.x->data(); - auto* dout = param.output; + auto& dout = param.output; auto in_dim = param.x->dims(); std::vector in_strides(in_dim.size()); in_strides[in_dim.size() - 1] = in_dim[in_dim.size() - 1]; diff --git a/paddle/fluid/lite/operators/op_params.h b/paddle/fluid/lite/operators/op_params.h index eee0d90dba2f3aad86a94983e0ac8fd67127b420..9986d199393d262ddf6aa9e54f6553fbb5e7627a 100644 --- a/paddle/fluid/lite/operators/op_params.h +++ b/paddle/fluid/lite/operators/op_params.h @@ -177,10 +177,10 @@ struct DropoutParam { // For Split op struct SplitParam { lite::Tensor* x{}; - std::vector* output{}; + std::vector output{}; int axis{-1}; int num{0}; - std::vector* sections; + std::vector sections; }; /// ----------------------- element wise operators ----------------------