diff options
Diffstat (limited to 'tests/benchmark/fixtures/PoolingLayerFixture.h')
| -rw-r--r-- | tests/benchmark/fixtures/PoolingLayerFixture.h | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/tests/benchmark/fixtures/PoolingLayerFixture.h b/tests/benchmark/fixtures/PoolingLayerFixture.h index 5a1a29612c..cbcfe2e869 100644 --- a/tests/benchmark/fixtures/PoolingLayerFixture.h +++ b/tests/benchmark/fixtures/PoolingLayerFixture.h @@ -48,7 +48,6 @@ public: void setup(TensorShape src_shape, PoolingLayerInfo info, DataType data_type, DataLayout data_layout, int batches) { // Set batched in source and destination shapes - const unsigned int fixed_point_position = 4; // Permute shape if NHWC format if(data_layout == DataLayout::NHWC) @@ -56,7 +55,7 @@ public: permute(src_shape, PermutationVector(2U, 0U, 1U)); } - TensorInfo src_info(src_shape, 1, data_type, fixed_point_position); + TensorInfo src_info(src_shape, 1, data_type); src_info.set_data_layout(data_layout); TensorShape dst_shape = compute_pool_shape(src_info, info); @@ -65,8 +64,8 @@ public: dst_shape.set(dst_shape.num_dimensions(), batches); // Create tensors - src = create_tensor<TensorType>(src_shape, data_type, 1, fixed_point_position, QuantizationInfo(), data_layout); - dst = create_tensor<TensorType>(dst_shape, data_type, 1, fixed_point_position, QuantizationInfo(), data_layout); + src = create_tensor<TensorType>(src_shape, data_type, 1, QuantizationInfo(), data_layout); + dst = create_tensor<TensorType>(dst_shape, data_type, 1, QuantizationInfo(), data_layout); // Create and configure function pool_layer.configure(&src, &dst, info); |