Browse Source

!15092 move kPerTensor field

From: @zhaodezan
Reviewed-by: @zhang_xue_tong,@jpc_chenjianping
Signed-off-by: @zhang_xue_tong
pull/15092/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
c1163d1abb
4 changed files with 4 additions and 3 deletions
  1. +1
    -0
      mindspore/lite/micro/coder/opcoders/base/conv2d_base_coder.cc
  2. +0
    -3
      mindspore/lite/src/lite_kernel.h
  3. +1
    -0
      mindspore/lite/src/runtime/kernel/arm/base/convolution_base.h
  4. +2
    -0
      mindspore/lite/src/weight_decoder.h

+ 1
- 0
mindspore/lite/micro/coder/opcoders/base/conv2d_base_coder.cc View File

@@ -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() {


+ 0
- 3
mindspore/lite/src/lite_kernel.h View File

@@ -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,


+ 1
- 0
mindspore/lite/src/runtime/kernel/arm/base/convolution_base.h View File

@@ -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;



+ 2
- 0
mindspore/lite/src/weight_decoder.h View File

@@ -27,6 +27,8 @@
#include "src/common/utils.h"
#include "src/tensor.h"

static constexpr int kPerTensor = 1;

namespace mindspore::lite {
class WeightDecoder {
public:


Loading…
Cancel
Save