!5036 fix leakyRelu parser bug

Merge pull request !5036 from lyvette/tflite_parser
This commit is contained in:
mindspore-ci-bot 2020-08-24 18:51:05 +08:00 committed by Gitee
commit 1dab825a1b
1 changed files with 1 additions and 1 deletions

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;