diff options
Diffstat (limited to 'tests/benchmark/fixtures/DepthwiseConvolutionLayerFixture.h')
-rw-r--r-- | tests/benchmark/fixtures/DepthwiseConvolutionLayerFixture.h | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/tests/benchmark/fixtures/DepthwiseConvolutionLayerFixture.h b/tests/benchmark/fixtures/DepthwiseConvolutionLayerFixture.h index 9276431de8..48ea03810f 100644 --- a/tests/benchmark/fixtures/DepthwiseConvolutionLayerFixture.h +++ b/tests/benchmark/fixtures/DepthwiseConvolutionLayerFixture.h @@ -57,15 +57,15 @@ public: weights_shape.set(2, dst_shape.z()); // Set batched in source and destination shapes - const unsigned int fixed_point_position = 4; + src_shape.set(3 /* batch */, batches); dst_shape.set(3 /* batch */, batches); // Create tensors - src = create_tensor<TensorType>(src_shape, data_type, 1, fixed_point_position, QuantizationInfo(0.5f, 10)); - weights = create_tensor<TensorType>(weights_shape, data_type, 1, fixed_point_position, QuantizationInfo(0.5f, 10)); - biases = create_tensor<TensorType>(TensorShape(weights_shape[2]), is_data_type_quantized_asymmetric(data_type) ? DataType::S32 : data_type, 1, fixed_point_position); - dst = create_tensor<TensorType>(dst_shape, data_type, 1, fixed_point_position, QuantizationInfo(0.5f, 10)); + src = create_tensor<TensorType>(src_shape, data_type, 1, QuantizationInfo(0.5f, 10)); + weights = create_tensor<TensorType>(weights_shape, data_type, 1, QuantizationInfo(0.5f, 10)); + biases = create_tensor<TensorType>(TensorShape(weights_shape[2]), is_data_type_quantized_asymmetric(data_type) ? DataType::S32 : data_type, 1); + dst = create_tensor<TensorType>(dst_shape, data_type, 1, QuantizationInfo(0.5f, 10)); // Create and configure function depth_conv.configure(&src, &weights, &biases, &dst, info); |