diff options
Diffstat (limited to 'src/armnnUtils/TensorIOUtils.hpp')
| -rw-r--r-- | src/armnnUtils/TensorIOUtils.hpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/armnnUtils/TensorIOUtils.hpp b/src/armnnUtils/TensorIOUtils.hpp index bf5a37b00f..47e0a320b8 100644 --- a/src/armnnUtils/TensorIOUtils.hpp +++ b/src/armnnUtils/TensorIOUtils.hpp @@ -37,7 +37,11 @@ inline armnn::InputTensors MakeInputTensors( { if (value.size() != inputBinding.second.GetNumElements()) { - throw armnn::Exception("Input tensor has incorrect size"); + std::ostringstream msg; + msg << "Input tensor has incorrect size (expected " + << inputBinding.second.GetNumElements() << " got " + << value.size(); + throw armnn::Exception(msg.str()); } armnn::ConstTensor inputTensor(inputBinding.second, value.data()); @@ -84,4 +88,4 @@ inline armnn::OutputTensors MakeOutputTensors( return outputTensors; } -} // namespace armnnUtils
\ No newline at end of file +} // namespace armnnUtils |