fix tflite parser bug

4 years ago · 57798b2efd
parent 9d2e07ae24
commit 57798b2efd
1 changed files with 1 additions and 1 deletions
--- 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_ptr<t
  attr->blockSize = tflite_attr->block_size;
  attr->format = schema::Format::Format_NHWC;
  auto primitive = std::make_unique<schema::PrimitiveT>();
-  primitive->value.type = schema::PrimitiveType_Concat;
+  primitive->value.type = schema::PrimitiveType_DepthToSpace;
  primitive->value.value = attr.release();
  return PrimitiveC::Create(primitive.release());
 }