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