diff options
Diffstat (limited to 'tests/benchmark/common')
-rw-r--r-- | tests/benchmark/common/ActivationLayer.h | 4 | ||||
-rw-r--r-- | tests/benchmark/common/ConvolutionLayer.h | 8 | ||||
-rw-r--r-- | tests/benchmark/common/FullyConnectedLayer.h | 8 | ||||
-rw-r--r-- | tests/benchmark/common/NormalizationLayer.h | 4 | ||||
-rw-r--r-- | tests/benchmark/common/PoolingLayer.h | 4 |
5 files changed, 14 insertions, 14 deletions
diff --git a/tests/benchmark/common/ActivationLayer.h b/tests/benchmark/common/ActivationLayer.h index 7edfb6ef3c..8e22281025 100644 --- a/tests/benchmark/common/ActivationLayer.h +++ b/tests/benchmark/common/ActivationLayer.h @@ -57,8 +57,8 @@ public: shape.set(shape.num_dimensions(), batches); // Create tensors - src = create_tensor(shape, dt, 1, fixed_point_position); - dst = create_tensor(shape, dt, 1, fixed_point_position); + src = create_tensor<TensorType>(shape, dt, 1, fixed_point_position); + dst = create_tensor<TensorType>(shape, dt, 1, fixed_point_position); // Create and configure function act_layer.configure(&src, &dst, act_obj.info); diff --git a/tests/benchmark/common/ConvolutionLayer.h b/tests/benchmark/common/ConvolutionLayer.h index 594c62c50e..a777a95bde 100644 --- a/tests/benchmark/common/ConvolutionLayer.h +++ b/tests/benchmark/common/ConvolutionLayer.h @@ -59,10 +59,10 @@ public: dst_shape.set(3 /* batch */, batches); // Create tensors - src = create_tensor(src_shape, dt, 1, fixed_point_position); - weights = create_tensor(conv_obj.weights_shape, dt, 1, fixed_point_position); - bias = create_tensor(conv_obj.bias_shape, dt, 1, fixed_point_position); - dst = create_tensor(dst_shape, dt, 1, fixed_point_position); + src = create_tensor<TensorType>(src_shape, dt, 1, fixed_point_position); + weights = create_tensor<TensorType>(conv_obj.weights_shape, dt, 1, fixed_point_position); + bias = create_tensor<TensorType>(conv_obj.bias_shape, dt, 1, fixed_point_position); + dst = create_tensor<TensorType>(dst_shape, dt, 1, fixed_point_position); // Create and configure function conv_layer = std::unique_ptr<Function>(new Function()); diff --git a/tests/benchmark/common/FullyConnectedLayer.h b/tests/benchmark/common/FullyConnectedLayer.h index 88adf83f2a..eeef1de28a 100644 --- a/tests/benchmark/common/FullyConnectedLayer.h +++ b/tests/benchmark/common/FullyConnectedLayer.h @@ -60,10 +60,10 @@ public: dst_shape.set(dst_shape.num_dimensions(), batches); // Create tensors - src = create_tensor(src_shape, dt, 1, fixed_point_position); - weights = create_tensor(fc_obj.weights_shape, dt, 1, fixed_point_position); - bias = create_tensor(fc_obj.bias_shape, dt, 1, fixed_point_position); - dst = create_tensor(dst_shape, dt, 1, fixed_point_position); + src = create_tensor<TensorType>(src_shape, dt, 1, fixed_point_position); + weights = create_tensor<TensorType>(fc_obj.weights_shape, dt, 1, fixed_point_position); + bias = create_tensor<TensorType>(fc_obj.bias_shape, dt, 1, fixed_point_position); + dst = create_tensor<TensorType>(dst_shape, dt, 1, fixed_point_position); // Create and configure function fc_layer = std::unique_ptr<Function>(new Function()); diff --git a/tests/benchmark/common/NormalizationLayer.h b/tests/benchmark/common/NormalizationLayer.h index 4593fb7df3..580dca6c48 100644 --- a/tests/benchmark/common/NormalizationLayer.h +++ b/tests/benchmark/common/NormalizationLayer.h @@ -58,8 +58,8 @@ public: shape.set(shape.num_dimensions(), batches); // Create tensors - src = create_tensor(shape, dt, 1, fixed_point_position); - dst = create_tensor(shape, dt, 1, fixed_point_position); + src = create_tensor<TensorType>(shape, dt, 1, fixed_point_position); + dst = create_tensor<TensorType>(shape, dt, 1, fixed_point_position); // Create and configure function norm_layer = std::unique_ptr<Function>(new Function()); diff --git a/tests/benchmark/common/PoolingLayer.h b/tests/benchmark/common/PoolingLayer.h index 5bb332fd6b..96ff905568 100644 --- a/tests/benchmark/common/PoolingLayer.h +++ b/tests/benchmark/common/PoolingLayer.h @@ -59,8 +59,8 @@ public: dst_shape.set(dst_shape.num_dimensions(), batches); // Create tensors - src = create_tensor(src_shape, dt, 1, fixed_point_position); - dst = create_tensor(dst_shape, dt, 1, fixed_point_position); + src = create_tensor<TensorType>(src_shape, dt, 1, fixed_point_position); + dst = create_tensor<TensorType>(dst_shape, dt, 1, fixed_point_position); // Create and configure function pool_layer.configure(&src, &dst, pool_obj.info); |