diff options
Diffstat (limited to 'tests/benchmark/CL/GEMM.h')
-rw-r--r-- | tests/benchmark/CL/GEMM.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/benchmark/CL/GEMM.h b/tests/benchmark/CL/GEMM.h index 02a339609c..ca3d9ad594 100644 --- a/tests/benchmark/CL/GEMM.h +++ b/tests/benchmark/CL/GEMM.h @@ -59,10 +59,10 @@ public: TensorShape shape_d = gemm_obj.shape_d; // Create tensors - a = create_tensor(shape_a, data_type); - b = create_tensor(shape_b, data_type); - c = create_tensor(shape_c, data_type); - d = create_tensor(shape_d, data_type); + a = create_tensor<CLTensor>(shape_a, data_type); + b = create_tensor<CLTensor>(shape_b, data_type); + c = create_tensor<CLTensor>(shape_c, data_type); + d = create_tensor<CLTensor>(shape_d, data_type); // Create and configure function gemm_layer = std::unique_ptr<Function>(new Function()); |