move kPerTensor field

This commit is contained in:
zhaodezan 2021-04-13 16:53:11 +08:00
parent 0310475fa9
commit e1b2688cfe
4 changed files with 4 additions and 3 deletions

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

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,

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;

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: