!8500 [MSLITE][Develop] fix code check

From: @sunsuodong
Reviewed-by: @zhanghaibo5,@zhang_xue_tong
Signed-off-by: @zhang_xue_tong
pull/8500/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit ce18db3f4c

@ -27,13 +27,12 @@ namespace mindspore {
namespace lite { namespace lite {
class Abs : public ArithmeticSelf { class Abs : public ArithmeticSelf {
public: public:
Abs() = default;
~Abs() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(Abs, ArithmeticSelf); MS_DECLARE_PARENT(Abs, ArithmeticSelf);
Abs() = default;
explicit Abs(schema::PrimitiveT *primitive) : ArithmeticSelf(primitive) {} explicit Abs(schema::PrimitiveT *primitive) : ArithmeticSelf(primitive) {}
#else #else
Abs() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
}; };

@ -26,9 +26,10 @@ namespace mindspore {
namespace lite { namespace lite {
class Activation : public PrimitiveC { class Activation : public PrimitiveC {
public: public:
Activation() = default;
~Activation() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(Activation, PrimitiveC); MS_DECLARE_PARENT(Activation, PrimitiveC);
Activation() = default;
explicit Activation(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit Activation(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
void SetType(int type); void SetType(int type);
@ -36,8 +37,6 @@ class Activation : public PrimitiveC {
void SetMinVal(float minVal); void SetMinVal(float minVal);
void SetMaxVal(float maxVal); void SetMaxVal(float maxVal);
#else #else
Activation() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int GetType() const; int GetType() const;

@ -28,16 +28,15 @@ namespace mindspore {
namespace lite { namespace lite {
class ActivationGrad : public PrimitiveC { class ActivationGrad : public PrimitiveC {
public: public:
ActivationGrad() = default;
~ActivationGrad() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(ActivationGrad, PrimitiveC); MS_DECLARE_PARENT(ActivationGrad, PrimitiveC);
ActivationGrad() = default;
explicit ActivationGrad(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit ActivationGrad(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
void SetType(int type); void SetType(int type);
void SetAlpha(float alpha); void SetAlpha(float alpha);
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
#else #else
ActivationGrad() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int GetType() const; int GetType() const;

@ -28,14 +28,13 @@ namespace mindspore {
namespace lite { namespace lite {
class Adam : public PrimitiveC { class Adam : public PrimitiveC {
public: public:
Adam() = default;
~Adam() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(Adam, PrimitiveC); MS_DECLARE_PARENT(Adam, PrimitiveC);
Adam() = default;
explicit Adam(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit Adam(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
#else #else
Adam() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override; int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override;

@ -26,15 +26,14 @@ namespace mindspore {
namespace lite { namespace lite {
class Add : public Arithmetic { class Add : public Arithmetic {
public: public:
Add() = default;
~Add() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(Add, Arithmetic); MS_DECLARE_PARENT(Add, Arithmetic);
Add() = default;
explicit Add(schema::PrimitiveT *primitive) : Arithmetic(primitive) {} explicit Add(schema::PrimitiveT *primitive) : Arithmetic(primitive) {}
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
void SetActivationType(int activation_type); void SetActivationType(int activation_type);
#else #else
Add() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int GetActivationType() const; int GetActivationType() const;

@ -26,15 +26,14 @@ namespace mindspore {
namespace lite { namespace lite {
class AddN : public PrimitiveC { class AddN : public PrimitiveC {
public: public:
AddN() = default;
~AddN() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(AddN, PrimitiveC); MS_DECLARE_PARENT(AddN, PrimitiveC);
AddN() = default;
explicit AddN(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit AddN(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
void SetN(int n); void SetN(int n);
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
#else #else
AddN() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override; int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override;

@ -28,14 +28,13 @@ namespace mindspore {
namespace lite { namespace lite {
class ApplyMomentum : public PrimitiveC { class ApplyMomentum : public PrimitiveC {
public: public:
ApplyMomentum() = default;
~ApplyMomentum() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(ApplyMomentum, PrimitiveC); MS_DECLARE_PARENT(ApplyMomentum, PrimitiveC);
ApplyMomentum() = default;
explicit ApplyMomentum(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit ApplyMomentum(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
#else #else
ApplyMomentum() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override; int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override;

@ -27,9 +27,10 @@ namespace mindspore {
namespace lite { namespace lite {
class ArgMax : public PrimitiveC { class ArgMax : public PrimitiveC {
public: public:
ArgMax() = default;
~ArgMax() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(ArgMax, PrimitiveC); MS_DECLARE_PARENT(ArgMax, PrimitiveC);
ArgMax() = default;
explicit ArgMax(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit ArgMax(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
void SetAxis(int axis); void SetAxis(int axis);
void SetOutMaxValue(bool out_max_value); void SetOutMaxValue(bool out_max_value);
@ -37,8 +38,6 @@ class ArgMax : public PrimitiveC {
void SetKeepDims(bool keep_dims); void SetKeepDims(bool keep_dims);
void SetAxisType(int axis_type); void SetAxisType(int axis_type);
#else #else
ArgMax() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override; int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override;

@ -26,9 +26,10 @@ namespace mindspore {
namespace lite { namespace lite {
class ArgMin : public PrimitiveC { class ArgMin : public PrimitiveC {
public: public:
ArgMin() = default;
~ArgMin() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(ArgMin, PrimitiveC); MS_DECLARE_PARENT(ArgMin, PrimitiveC);
ArgMin() = default;
explicit ArgMin(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit ArgMin(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
void SetAxis(int axis); void SetAxis(int axis);
void SetOutMaxValue(bool out_max_value); void SetOutMaxValue(bool out_max_value);
@ -36,8 +37,6 @@ class ArgMin : public PrimitiveC {
void SetKeepDims(bool keep_dims); void SetKeepDims(bool keep_dims);
void SetAxisType(int axis_type); void SetAxisType(int axis_type);
#else #else
ArgMin() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override; int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override;

@ -27,13 +27,13 @@ namespace mindspore {
namespace lite { namespace lite {
class Arithmetic : public PrimitiveC { class Arithmetic : public PrimitiveC {
public: public:
Arithmetic() = default;
~Arithmetic() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(Arithmetic, PrimitiveC); MS_DECLARE_PARENT(Arithmetic, PrimitiveC);
Arithmetic() = default;
explicit Arithmetic(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit Arithmetic(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
#else #else
// explicit Arithmetic(schema::Primitive *primitive) : PrimitiveC(primitive) {} // explicit Arithmetic(schema::Primitive *primitive) : PrimitiveC(primitive) {}
Arithmetic() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override { int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override {
return RET_ERROR; return RET_ERROR;
} }

@ -27,13 +27,13 @@ namespace mindspore {
namespace lite { namespace lite {
class ArithmeticGrad : public PrimitiveC { class ArithmeticGrad : public PrimitiveC {
public: public:
ArithmeticGrad() = default;
~ArithmeticGrad() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(ArithmeticGrad, PrimitiveC); MS_DECLARE_PARENT(ArithmeticGrad, PrimitiveC);
ArithmeticGrad() = default;
explicit ArithmeticGrad(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit ArithmeticGrad(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
#else #else
// explicit ArithmeticGrad(const schema::Primitive &primitive) : PrimitiveC(primitive) {} // explicit ArithmeticGrad(const schema::Primitive &primitive) : PrimitiveC(primitive) {}
ArithmeticGrad() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override { int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override {
return RET_ERROR; return RET_ERROR;
} }

@ -25,13 +25,13 @@ namespace mindspore {
namespace lite { namespace lite {
class ArithmeticSelf : public PrimitiveC { class ArithmeticSelf : public PrimitiveC {
public: public:
ArithmeticSelf() = default;
~ArithmeticSelf() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(ArithmeticSelf, PrimitiveC); MS_DECLARE_PARENT(ArithmeticSelf, PrimitiveC);
ArithmeticSelf() = default;
explicit ArithmeticSelf(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit ArithmeticSelf(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
#else #else
// explicit ArithmeticSelf(schema::Primitive *primitive) : PrimitiveC(primitive) {} // explicit ArithmeticSelf(schema::Primitive *primitive) : PrimitiveC(primitive) {}
ArithmeticSelf() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override { int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override {
return RET_ERROR; return RET_ERROR;
} }

@ -26,13 +26,13 @@ namespace mindspore {
namespace lite { namespace lite {
class Assign : public PrimitiveC { class Assign : public PrimitiveC {
public: public:
Assign() = default;
~Assign() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(Assign, PrimitiveC); MS_DECLARE_PARENT(Assign, PrimitiveC);
Assign() = default;
explicit Assign(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit Assign(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
#else #else
Assign() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override; int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override;

@ -24,14 +24,13 @@ namespace mindspore {
namespace lite { namespace lite {
class AssignAdd : public PrimitiveC { class AssignAdd : public PrimitiveC {
public: public:
AssignAdd() = default;
~AssignAdd() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(AssignAdd, PrimitiveC); MS_DECLARE_PARENT(AssignAdd, PrimitiveC);
AssignAdd() = default;
explicit AssignAdd(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit AssignAdd(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
#else #else
AssignAdd() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override; int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override;

@ -26,16 +26,15 @@ namespace mindspore {
namespace lite { namespace lite {
class AudioSpectrogram : public PrimitiveC { class AudioSpectrogram : public PrimitiveC {
public: public:
AudioSpectrogram() = default;
~AudioSpectrogram() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(AudioSpectrogram, PrimitiveC); MS_DECLARE_PARENT(AudioSpectrogram, PrimitiveC);
AudioSpectrogram() = default;
explicit AudioSpectrogram(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit AudioSpectrogram(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
void SetWindowSize(int window_size) { this->primitive_->value.AsAudioSpectrogram()->windowSize = window_size; } void SetWindowSize(int window_size) { this->primitive_->value.AsAudioSpectrogram()->windowSize = window_size; }
void SetStride(int stride) { this->primitive_->value.AsAudioSpectrogram()->stride = stride; } void SetStride(int stride) { this->primitive_->value.AsAudioSpectrogram()->stride = stride; }
void SetMagSquare(bool mag_square) { this->primitive_->value.AsAudioSpectrogram()->magSquare = mag_square; } void SetMagSquare(bool mag_square) { this->primitive_->value.AsAudioSpectrogram()->magSquare = mag_square; }
#else #else
AudioSpectrogram() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int GetWindowSize() const; int GetWindowSize() const;

@ -26,15 +26,14 @@ namespace mindspore {
namespace lite { namespace lite {
class BatchNorm : public PrimitiveC { class BatchNorm : public PrimitiveC {
public: public:
BatchNorm() = default;
~BatchNorm() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(BatchNorm, PrimitiveC); MS_DECLARE_PARENT(BatchNorm, PrimitiveC);
BatchNorm() = default;
explicit BatchNorm(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit BatchNorm(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
void SetEpsilon(float epsilon); void SetEpsilon(float epsilon);
#else #else
BatchNorm() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
float GetEpsilon() const; float GetEpsilon() const;

@ -27,15 +27,14 @@ namespace mindspore {
namespace lite { namespace lite {
class BatchToSpace : public PrimitiveC { class BatchToSpace : public PrimitiveC {
public: public:
BatchToSpace() = default;
~BatchToSpace() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(BatchToSpace, PrimitiveC); MS_DECLARE_PARENT(BatchToSpace, PrimitiveC);
BatchToSpace() = default;
explicit BatchToSpace(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit BatchToSpace(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
void SetBlockShape(const std::vector<int> &block_shape); void SetBlockShape(const std::vector<int> &block_shape);
void SetCrops(const std::vector<int> &crops); void SetCrops(const std::vector<int> &crops);
#else #else
BatchToSpace() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override; int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override;

@ -27,15 +27,14 @@ namespace mindspore {
namespace lite { namespace lite {
class BiasAdd : public PrimitiveC { class BiasAdd : public PrimitiveC {
public: public:
BiasAdd() = default;
~BiasAdd() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(BiasAdd, PrimitiveC); MS_DECLARE_PARENT(BiasAdd, PrimitiveC);
BiasAdd() = default;
explicit BiasAdd(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit BiasAdd(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
void SetAxis(const std::vector<int> &axis); void SetAxis(const std::vector<int> &axis);
#else #else
BiasAdd() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
std::vector<int> GetAxis() const; std::vector<int> GetAxis() const;

@ -27,15 +27,14 @@ namespace mindspore {
namespace lite { namespace lite {
class BiasGrad : public PrimitiveC { class BiasGrad : public PrimitiveC {
public: public:
BiasGrad() = default;
~BiasGrad() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(BiasGrad, PrimitiveC); MS_DECLARE_PARENT(BiasGrad, PrimitiveC);
BiasGrad() = default;
explicit BiasGrad(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit BiasGrad(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
void SetAxis(const std::vector<int> &axis); void SetAxis(const std::vector<int> &axis);
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
#else #else
BiasGrad() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int InferShape(std::vector<lite::Tensor *> inputs, std::vector<lite::Tensor *> outputs) override; int InferShape(std::vector<lite::Tensor *> inputs, std::vector<lite::Tensor *> outputs) override;

@ -25,19 +25,17 @@ namespace mindspore {
namespace lite { namespace lite {
class BinaryCrossEntropy : public PrimitiveC { class BinaryCrossEntropy : public PrimitiveC {
public: public:
BinaryCrossEntropy() = default;
~BinaryCrossEntropy() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(BinaryCrossEntropy, PrimitiveC); MS_DECLARE_PARENT(BinaryCrossEntropy, PrimitiveC);
BinaryCrossEntropy() = default;
explicit BinaryCrossEntropy(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit BinaryCrossEntropy(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
int GetReduction() const; int GetReduction() const;
#else #else
BinaryCrossEntropy() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
int GetReduction() const; int GetReduction() const;

@ -25,19 +25,17 @@ namespace mindspore {
namespace lite { namespace lite {
class BinaryCrossEntropyGrad : public PrimitiveC { class BinaryCrossEntropyGrad : public PrimitiveC {
public: public:
BinaryCrossEntropyGrad() = default;
~BinaryCrossEntropyGrad() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(BinaryCrossEntropyGrad, PrimitiveC); MS_DECLARE_PARENT(BinaryCrossEntropyGrad, PrimitiveC);
BinaryCrossEntropyGrad() = default;
explicit BinaryCrossEntropyGrad(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit BinaryCrossEntropyGrad(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
int GetReduction() const; int GetReduction() const;
#else #else
BinaryCrossEntropyGrad() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
int GetReduction() const; int GetReduction() const;

@ -26,15 +26,15 @@ namespace mindspore {
namespace lite { namespace lite {
class BNGrad : public PrimitiveC { class BNGrad : public PrimitiveC {
public: public:
BNGrad() = default;
~BNGrad() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(BNGrad, PrimitiveC); MS_DECLARE_PARENT(BNGrad, PrimitiveC);
BNGrad() = default;
explicit BNGrad(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit BNGrad(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
void SetEps(float eps); void SetEps(float eps);
void SetMomentum(float momentum); void SetMomentum(float momentum);
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override; int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) override;
#else #else
BNGrad() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override; int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override;

@ -27,15 +27,14 @@ namespace mindspore {
namespace lite { namespace lite {
class BroadcastTo : public PrimitiveC { class BroadcastTo : public PrimitiveC {
public: public:
BroadcastTo() = default;
~BroadcastTo() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(BroadcastTo, PrimitiveC); MS_DECLARE_PARENT(BroadcastTo, PrimitiveC);
BroadcastTo() = default;
explicit BroadcastTo(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit BroadcastTo(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
void SetDstShape(const std::vector<int> &dst_shape); void SetDstShape(const std::vector<int> &dst_shape);
#else #else
BroadcastTo() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override; int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override;

@ -26,16 +26,15 @@ namespace mindspore {
namespace lite { namespace lite {
class Cast : public PrimitiveC { class Cast : public PrimitiveC {
public: public:
Cast() = default;
~Cast() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(Cast, PrimitiveC); MS_DECLARE_PARENT(Cast, PrimitiveC);
Cast() = default;
explicit Cast(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {} explicit Cast(schema::PrimitiveT *primitive) : PrimitiveC(primitive) {}
void SetSrcT(int src_t); void SetSrcT(int src_t);
void SetDstT(int dst_t); void SetDstT(int dst_t);
int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs); int UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs);
#else #else
Cast() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override; int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override;
#endif #endif
int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override; int InferShape(std::vector<lite::Tensor *> inputs_, std::vector<lite::Tensor *> outputs_) override;

@ -26,13 +26,12 @@ namespace mindspore {
namespace lite { namespace lite {
class Ceil : public ArithmeticSelf { class Ceil : public ArithmeticSelf {
public: public:
Ceil() = default;
~Ceil() = default;
#ifdef PRIMITIVE_WRITEABLE #ifdef PRIMITIVE_WRITEABLE
MS_DECLARE_PARENT(Ceil, ArithmeticSelf); MS_DECLARE_PARENT(Ceil, ArithmeticSelf);
Ceil() = default;
explicit Ceil(schema::PrimitiveT *primitive) : ArithmeticSelf(primitive) {} explicit Ceil(schema::PrimitiveT *primitive) : ArithmeticSelf(primitive) {}
#else #else
Ceil() = default;
int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override { int UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) override {
MS_ASSERT(nullptr != primitive); MS_ASSERT(nullptr != primitive);
MS_ASSERT(nullptr != fbb); MS_ASSERT(nullptr != fbb);

Some files were not shown because too many files have changed in this diff Show More

Loading…
Cancel
Save