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