diff options
Diffstat (limited to 'include/armnnTestUtils/MockTensorHandle.hpp')
-rw-r--r-- | include/armnnTestUtils/MockTensorHandle.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/include/armnnTestUtils/MockTensorHandle.hpp b/include/armnnTestUtils/MockTensorHandle.hpp index 9a7518b21a..bc2f7d7e3b 100644 --- a/include/armnnTestUtils/MockTensorHandle.hpp +++ b/include/armnnTestUtils/MockTensorHandle.hpp @@ -5,7 +5,13 @@ #pragma once #include "MockMemoryManager.hpp" + #include <armnn/backends/TensorHandle.hpp> +#include <armnn/MemorySources.hpp> +#include <armnn/Tensor.hpp> +#include <armnn/Types.hpp> +#include <armnn/backends/ITensorHandle.hpp> +#include <memory> namespace armnn { |