diff --git a/mindspore/lite/src/train/train_populate_parameter.cc b/mindspore/lite/src/train/train_populate_parameter.cc index 12bca1b58b..0d77cb0583 100644 --- a/mindspore/lite/src/train/train_populate_parameter.cc +++ b/mindspore/lite/src/train/train_populate_parameter.cc @@ -82,6 +82,7 @@ OpParameter *PopulateBCEParameter(const void *prim) { } auto primitive = static_cast(prim); auto value = primitive->value_as_BinaryCrossEntropy(); + MS_ASSERT(value != nullptr); *reduction = value->reduction(); return reinterpret_cast(reduction); } diff --git a/mindspore/lite/tools/converter/parser/tflite/tflite_logical_parser.h b/mindspore/lite/tools/converter/parser/tflite/tflite_logical_parser.h index cda957851d..aa530a0580 100644 --- a/mindspore/lite/tools/converter/parser/tflite/tflite_logical_parser.h +++ b/mindspore/lite/tools/converter/parser/tflite/tflite_logical_parser.h @@ -39,6 +39,8 @@ class TfliteLogicalNotParser : public TfliteNodeParser { public: TfliteLogicalNotParser() : TfliteNodeParser("LogicalNot") {} + ~TfliteLogicalNotParser() override = default; + ops::PrimitiveC *Parse(const std::unique_ptr &tflite_op, const std::unique_ptr &tflite_model) override; }; @@ -47,6 +49,8 @@ class TfliteLogicalOrParser : public TfliteNodeParser { public: TfliteLogicalOrParser() : TfliteNodeParser("LogicalOr") {} + ~TfliteLogicalOrParser() override = default; + ops::PrimitiveC *Parse(const std::unique_ptr &tflite_op, const std::unique_ptr &tflite_model) override; };