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()); }