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 value = primitive->value_as_BinaryCrossEntropy();
MS_ASSERT(value != nullptr);
*reduction = value->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:
TfliteLogicalNotParser() : TfliteNodeParser("LogicalNot") {}

~TfliteLogicalNotParser() override = default;

ops::PrimitiveC *Parse(const std::unique_ptr<tflite::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &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::OperatorT> &tflite_op,
const std::unique_ptr<tflite::ModelT> &tflite_model) override;
};


Loading…
Cancel
Save