Browse Source

!12388 [MS][LITE] fix tflite parser bug

From: @cjh9368
Reviewed-by: @zhanghaibo5,@hangangqiang
Signed-off-by: @zhanghaibo5
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
eee7374120
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/tools/converter/parser/tflite/tflite_depth_to_space_parser.cc

+ 1
- 1
mindspore/lite/tools/converter/parser/tflite/tflite_depth_to_space_parser.cc View File

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


Loading…
Cancel
Save