Browse Source

!4036 fix tflite transpose parser bug

Merge pull request !4036 from wangzhe/master
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
443d7a9133
1 changed files with 1 additions and 5 deletions
  1. +1
    -5
      mindspore/lite/tools/converter/parser/tflite/tflite_transpose_parser.cc

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

@@ -27,13 +27,9 @@ STATUS TfliteTransposeParser::Parse(const std::unique_ptr<tflite::OperatorT> &tf
schema::CNodeT *op, TensorCache *tensor_cache, bool quantizedModel) {
MS_LOG(DEBUG) << "parse TfliteTransposeParser";
std::unique_ptr<schema::TransposeT> attr(new schema::TransposeT());
const auto &tfliteAttr = tfliteOp->builtin_options.AsTransposeOptions();
if (tfliteAttr == nullptr) {
MS_LOG(ERROR) << "get op: " << op->name.c_str() << " attr failed";
return RET_NULL_PTR;
}

if (GetTfliteData(tfliteOp->inputs[1], tfliteTensors, tfliteModelBuffer, attr->perm)) {
MS_LOG(ERROR) << "parse Transpose attr perm failed";
return RET_ERROR;
}



Loading…
Cancel
Save