OSSInnovation
/
mindspore
1
0
Fork 26
Code Issues Pull Requests 7 Releases Wiki Activity

fix tflite parser bug

This commit is contained in:
cjh9368 2021-02-18 17:07:33 +08:00
parent 9d2e07ae24
commit 57798b2efd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
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->blockSize = tflite_attr->block_size;
attr->format = schema::Format::Format_NHWC; attr->format = schema::Format::Format_NHWC;
auto primitive = std::make_unique<schema::PrimitiveT>(); auto primitive = std::make_unique<schema::PrimitiveT>();
primitive->value.type = schema::PrimitiveType_Concat; primitive->value.type = schema::PrimitiveType_DepthToSpace;
primitive->value.value = attr.release(); primitive->value.value = attr.release();
return PrimitiveC::Create(primitive.release()); return PrimitiveC::Create(primitive.release());
} }