!5036 fix leakyRelu parser bug
Merge pull request !5036 from lyvette/tflite_parser
This commit is contained in:
commit
1dab825a1b
|
@ -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…
Reference in New Issue