From aa14eab63d7fbb6f6ef5ede37431f9f9edfe5823 Mon Sep 17 00:00:00 2001 From: yefeng Date: Fri, 30 Apr 2021 14:58:28 +0800 Subject: [PATCH] codedex_master --- mindspore/lite/src/train/train_populate_parameter.cc | 1 + .../tools/converter/parser/tflite/tflite_logical_parser.h | 4 ++++ 2 files changed, 5 insertions(+) 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; };