diff --git a/ethosu/regor/tflite/tflite_supported_operators.cpp b/ethosu/regor/tflite/tflite_supported_operators.cpp index e5454b26f14cafd0c9d0d202b8dad26cde541506..6f47222156bf02894948b65cd291d35876935a3f 100644 --- a/ethosu/regor/tflite/tflite_supported_operators.cpp +++ b/ethosu/regor/tflite/tflite_supported_operators.cpp @@ -727,13 +727,19 @@ void TfLiteSupportedOperators::Failure(const Operation *op, const std::string &m assert(op); auto ofmConn = op->Output(TensorUsage::OFM); const char *name = "N/A"; + OpType opType = op->Type(); if ( ofmConn && ofmConn->tensor ) { name = ofmConn->tensor->Name().c_str(); } - auto tfLiteType = TfLiteMapping::OpTypeToBuiltinOperator(op->Type()); + std::string type = OpTypeToString(op->Type()); + if ( opType != OpType::None ) + { + auto tfLiteType = TfLiteMapping::OpTypeToBuiltinOperator(opType); + type = TfLiteMapping::BuiltinOperatorToString(tfLiteType); + } assert(message.size() || constraint.size()); - LOG_WARN("\nWarning (supported operators) operator:{} ofm:{}\n", TfLiteMapping::BuiltinOperatorToString(tfLiteType), name); + LOG_WARN("\nWarning (supported operators) operator:{} ofm:{}\n", std::move(type), name); if ( message.size() ) { LOG_WARN("Reason: {}\n", message);