diff --git a/mindspore/lite/micro/coder/opcoders/base/conv2d_base_coder.cc b/mindspore/lite/micro/coder/opcoders/base/conv2d_base_coder.cc index 7735f5711fe..46be375effd 100644 --- a/mindspore/lite/micro/coder/opcoders/base/conv2d_base_coder.cc +++ b/mindspore/lite/micro/coder/opcoders/base/conv2d_base_coder.cc @@ -20,6 +20,7 @@ #include "nnacl/fp32/winograd_utils.h" #include "nnacl/int8/quantize.h" #include "coder/log.h" +#include "src/weight_decoder.h" namespace mindspore::lite::micro { Conv2DBaseCoder::~Conv2DBaseCoder() { diff --git a/mindspore/lite/src/lite_kernel.h b/mindspore/lite/src/lite_kernel.h index 7a4747dd4d0..785c9d49401 100644 --- a/mindspore/lite/src/lite_kernel.h +++ b/mindspore/lite/src/lite_kernel.h @@ -31,9 +31,6 @@ #include "include/errorcode.h" #include "schema/model_generated.h" -static constexpr int kPerTensor = 1; -static constexpr size_t kPerBatch = 3; - namespace mindspore::kernel { enum KERNEL_ARCH { kCPU, diff --git a/mindspore/lite/src/runtime/kernel/arm/base/convolution_base.h b/mindspore/lite/src/runtime/kernel/arm/base/convolution_base.h index 8f6d61914a5..4793e85a33c 100644 --- a/mindspore/lite/src/runtime/kernel/arm/base/convolution_base.h +++ b/mindspore/lite/src/runtime/kernel/arm/base/convolution_base.h @@ -30,6 +30,7 @@ #include "src/lite_kernel.h" #include "include/context.h" #include "src/runtime/kernel/arm/base/layout_transform.h" +#include "src/weight_decoder.h" using mindspore::lite::InnerContext; diff --git a/mindspore/lite/src/weight_decoder.h b/mindspore/lite/src/weight_decoder.h index d7dd90c47de..b8baa2f036f 100644 --- a/mindspore/lite/src/weight_decoder.h +++ b/mindspore/lite/src/weight_decoder.h @@ -27,6 +27,8 @@ #include "src/common/utils.h" #include "src/tensor.h" +static constexpr int kPerTensor = 1; + namespace mindspore::lite { class WeightDecoder { public: