diff options
Diffstat (limited to 'src/backends/reference/test')
-rw-r--r-- | src/backends/reference/test/RefLayerSupportTests.cpp | 2 | ||||
-rw-r--r-- | src/backends/reference/test/RefTensorHandleTests.cpp | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/backends/reference/test/RefLayerSupportTests.cpp b/src/backends/reference/test/RefLayerSupportTests.cpp index 1648583210..a1487061b6 100644 --- a/src/backends/reference/test/RefLayerSupportTests.cpp +++ b/src/backends/reference/test/RefLayerSupportTests.cpp @@ -7,7 +7,7 @@ #include <layers/ConvertFp32ToFp16Layer.hpp> #include <test/TensorHelpers.hpp> -#include <backendsCommon/CpuTensorHandle.hpp> +#include <backendsCommon/TensorHandle.hpp> #include <reference/RefWorkloadFactory.hpp> #include <reference/RefLayerSupport.hpp> #include <backendsCommon/test/LayerTests.hpp> diff --git a/src/backends/reference/test/RefTensorHandleTests.cpp b/src/backends/reference/test/RefTensorHandleTests.cpp index b04d9d6c52..dadd1de1f2 100644 --- a/src/backends/reference/test/RefTensorHandleTests.cpp +++ b/src/backends/reference/test/RefTensorHandleTests.cpp @@ -173,8 +173,8 @@ BOOST_AUTO_TEST_CASE(TestManagedConstTensorHandle) void* mem = nullptr; TensorInfo info; - // Use PassthroughCpuTensor as others are abstract - auto passThroughHandle = std::make_shared<PassthroughCpuTensorHandle>(info, mem); + // Use PassthroughTensor as others are abstract + auto passThroughHandle = std::make_shared<PassthroughTensorHandle>(info, mem); // Test managed handle is initialized with m_Mapped unset and once Map() called its set ManagedConstTensorHandle managedHandle(passThroughHandle); |