Browse Source

!15959 [MS_LITE] codedex

From: @YeFeng_24
Reviewed-by: @hangangqiang,@zhanghaibo5
Signed-off-by: @hangangqiang
pull/15959/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
a492c6e89f
2 changed files with 5 additions and 0 deletions
  1. +1
    -0
      mindspore/lite/src/train/train_populate_parameter.cc
  2. +4
    -0
      mindspore/lite/tools/converter/parser/tflite/tflite_logical_parser.h

+ 1
- 0
mindspore/lite/src/train/train_populate_parameter.cc View File

@@ -82,6 +82,7 @@ OpParameter *PopulateBCEParameter(const void *prim) {
} }
auto primitive = static_cast<const schema::Primitive *>(prim); auto primitive = static_cast<const schema::Primitive *>(prim);
auto value = primitive->value_as_BinaryCrossEntropy(); auto value = primitive->value_as_BinaryCrossEntropy();
MS_ASSERT(value != nullptr);
*reduction = value->reduction(); *reduction = value->reduction();
return reinterpret_cast<OpParameter *>(reduction); return reinterpret_cast<OpParameter *>(reduction);
} }


+ 4
- 0
mindspore/lite/tools/converter/parser/tflite/tflite_logical_parser.h View File

@@ -39,6 +39,8 @@ class TfliteLogicalNotParser : public TfliteNodeParser {
public: public:
TfliteLogicalNotParser() : TfliteNodeParser("LogicalNot") {} TfliteLogicalNotParser() : TfliteNodeParser("LogicalNot") {}


~TfliteLogicalNotParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };
@@ -47,6 +49,8 @@ class TfliteLogicalOrParser : public TfliteNodeParser {
public: public:
TfliteLogicalOrParser() : TfliteNodeParser("LogicalOr") {} TfliteLogicalOrParser() : TfliteNodeParser("LogicalOr") {}


~TfliteLogicalOrParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op, ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override; const std::unique_ptr<tflite::ModelT> &tflite_model) override;
}; };


Loading…
Cancel
Save