diff --git a/lib/Onnxifi/Base.cpp b/lib/Onnxifi/Base.cpp index 7d71c33428..fc8028b64e 100644 --- a/lib/Onnxifi/Base.cpp +++ b/lib/Onnxifi/Base.cpp @@ -157,8 +157,18 @@ onnxStatus Graph::setIOAndRun(uint32_t inputsCount, } if (inOnnxTensorSize > inPhPtr->getType()->size()) { + std::stringstream ss; + for (const auto j : inOnnxTensorDims) { + ss << j << ", "; + } + ss << " vs "; + auto sizes = inPhPtr->getType()->dims(); + for (const auto j : sizes) { + ss << j << ", "; + } LOG(ERROR) << "Input tensor is too large: " << inOnnxTensorSize << " vs " - << inPhPtr->getType()->size() << ": " << inOnnxTensor.name; + << inPhPtr->getType()->size() << ": " << inOnnxTensor.name + << ", shape: " << ss.str(); return ONNXIFI_STATUS_INVALID_SHAPE; }