|
|
|
@@ -297,6 +297,12 @@ STATUS TfliteModelParser::ConvertGroupDepthwiseOp(schema::MetaGraphT *sub_graph) |
|
|
|
MS_LOG(ERROR) << "Trans depthwiseConv Filter schema::Format failed."; |
|
|
|
return RET_ERROR; |
|
|
|
} |
|
|
|
} else if (weight_tensor->dataType == kNumberTypeInt8) { |
|
|
|
auto status = TransFilterFormat<int8_t>(weight_tensor.get(), kKHWC2CHWK); |
|
|
|
if (status != RET_OK) { |
|
|
|
MS_LOG(ERROR) << "Trans filter format failed."; |
|
|
|
return RET_ERROR; |
|
|
|
} |
|
|
|
} else if (weight_tensor->dataType == kNumberTypeFloat32 || weight_tensor->dataType == kNumberTypeFloat) { |
|
|
|
auto status = TransFilterFormat<float>(weight_tensor.get(), kKHWC2CHWK); |
|
|
|
if (status != RET_OK) { |
|
|
|
|