From 57798b2efd84390c688ffbecc8b9118a82dc7da5 Mon Sep 17 00:00:00 2001 From: cjh9368 Date: Thu, 18 Feb 2021 17:07:33 +0800 Subject: [PATCH] fix tflite parser bug --- .../converter/parser/tflite/tflite_depth_to_space_parser.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/lite/tools/converter/parser/tflite/tflite_depth_to_space_parser.cc b/mindspore/lite/tools/converter/parser/tflite/tflite_depth_to_space_parser.cc index c8854a64f1..9292151208 100644 --- a/mindspore/lite/tools/converter/parser/tflite/tflite_depth_to_space_parser.cc +++ b/mindspore/lite/tools/converter/parser/tflite/tflite_depth_to_space_parser.cc @@ -37,7 +37,7 @@ PrimitiveC *TfliteDepthToSpaceParser::ParseLitePrimitive(const std::unique_ptrblockSize = tflite_attr->block_size; attr->format = schema::Format::Format_NHWC; auto primitive = std::make_unique(); - primitive->value.type = schema::PrimitiveType_Concat; + primitive->value.type = schema::PrimitiveType_DepthToSpace; primitive->value.value = attr.release(); return PrimitiveC::Create(primitive.release()); }