move kPerTensor field
This commit is contained in:
parent
0310475fa9
commit
e1b2688cfe
|
@ -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() {
|
||||
|
|
|
@ -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,
|
||||
|
|
|
@ -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;
|
||||
|
||||
|
|
|
@ -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…
Reference in New Issue