Browse Source

codedex_master

pull/15959/head
yefeng 5 years ago
parent
commit
aa14eab63d
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