diff options
author | Matthew Bentham <Matthew.Bentham@arm.com> | 2019-06-18 16:14:34 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-07-02 09:19:01 +0000 |
commit | 4cefc4135f7fbf5c2ba532a4f5b14d2811f9ed9e (patch) | |
tree | 1a7fbd6c636d83737b4e2319ed27255397921a32 /src/armnnUtils | |
parent | fe15eff9ed2007cd10800ec356ce2e8ea4b5f9d0 (diff) | |
download | armnn-4cefc4135f7fbf5c2ba532a4f5b14d2811f9ed9e.tar.gz |
IVGCVSW-3307 Introduce RefTensorHandle
Use it for intermediate tensors on reference backend.
Lays the groundwork for memory management in the reference backend.
Change-Id: I7d3ee132cac31bde70ae6e1b815f4f0b03d550a6
Signed-off-by: Matthew Bentham <Matthew.Bentham@arm.com>
Diffstat (limited to 'src/armnnUtils')
-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 |