Browse Source

fix leaky relu parser bug.

tags/v1.0.0
lyvette 5 years ago
parent
commit
2f04eda83e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/tools/converter/parser/tflite/tflite_activation_parser.cc

+ 1
- 1
mindspore/lite/tools/converter/parser/tflite/tflite_activation_parser.cc View File

@@ -71,7 +71,7 @@ STATUS TfliteActivationParser::Parse(const std::unique_ptr<tflite::OperatorT> &t
return RET_NULL_PTR;
}
attr->alpha = tflite_attr->alpha;
attr->type = schema::ActivationType_SIGMOID;
attr->type = schema::ActivationType_LEAKY_RELU;
}

op->primitive->value.type = schema::PrimitiveType_Activation;


Loading…
Cancel
Save