fix leaky relu parser bug.

2f04eda8 · lyvette · 6763b63c · 2f04eda8
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

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

未找到文件。
--- a/mindspore/lite/tools/converter/parser/tflite/tflite_activation_parser.cc
+++ b/mindspore/lite/tools/converter/parser/tflite/tflite_activation_parser.cc
@@ -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;