Browse Source

fix tflite parser bug

tags/v1.2.0-rc1
cjh9368 4 years ago
parent
commit
57798b2efd
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