aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/test/OpenClTimerTest.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/cl/test/OpenClTimerTest.cpp')
-rw-r--r--src/backends/cl/test/OpenClTimerTest.cpp10
1 files changed, 5 insertions, 5 deletions
diff --git a/src/backends/cl/test/OpenClTimerTest.cpp b/src/backends/cl/test/OpenClTimerTest.cpp
index 0e1f28ec4e..1b86d2e304 100644
--- a/src/backends/cl/test/OpenClTimerTest.cpp
+++ b/src/backends/cl/test/OpenClTimerTest.cpp
@@ -9,7 +9,7 @@
#include <test/TensorHelpers.hpp>
-#include <backendsCommon/CpuTensorHandle.hpp>
+#include <backendsCommon/TensorHandle.hpp>
#include <backendsCommon/WorkloadFactory.hpp>
#include <cl/ClContextControl.hpp>
@@ -79,10 +79,10 @@ BOOST_AUTO_TEST_CASE(OpenClTimerBatchNorm)
BatchNormalizationQueueDescriptor data;
WorkloadInfo info;
- ScopedCpuTensorHandle meanTensor(tensorInfo);
- ScopedCpuTensorHandle varianceTensor(tensorInfo);
- ScopedCpuTensorHandle betaTensor(tensorInfo);
- ScopedCpuTensorHandle gammaTensor(tensorInfo);
+ ScopedTensorHandle meanTensor(tensorInfo);
+ ScopedTensorHandle varianceTensor(tensorInfo);
+ ScopedTensorHandle betaTensor(tensorInfo);
+ ScopedTensorHandle gammaTensor(tensorInfo);
AllocateAndCopyDataToITensorHandle(&meanTensor, &mean[0]);
AllocateAndCopyDataToITensorHandle(&varianceTensor, &variance[0]);