diff options
Diffstat (limited to 'tests/benchmark/fixtures/BatchNormalizationLayerFixture.h')
-rw-r--r-- | tests/benchmark/fixtures/BatchNormalizationLayerFixture.h | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/tests/benchmark/fixtures/BatchNormalizationLayerFixture.h b/tests/benchmark/fixtures/BatchNormalizationLayerFixture.h index ae8f8a7af1..42adefeef6 100644 --- a/tests/benchmark/fixtures/BatchNormalizationLayerFixture.h +++ b/tests/benchmark/fixtures/BatchNormalizationLayerFixture.h @@ -45,7 +45,7 @@ public: void setup(TensorShape tensor_shape, TensorShape param_shape, float epsilon, bool use_gamma, bool use_beta, ActivationLayerInfo act_info, DataType data_type, DataLayout data_layout, int batches) { // Set batched in source and destination shapes - const unsigned int fixed_point_position = 4; + tensor_shape.set(tensor_shape.num_dimensions(), batches); if(data_layout == DataLayout::NHWC) { @@ -53,12 +53,12 @@ public: } // Create tensors - src = create_tensor<TensorType>(tensor_shape, data_type, 1, fixed_point_position, QuantizationInfo(), data_layout); - dst = create_tensor<TensorType>(tensor_shape, data_type, 1, fixed_point_position, QuantizationInfo(), data_layout); - mean = create_tensor<TensorType>(param_shape, data_type, 1, fixed_point_position); - variance = create_tensor<TensorType>(param_shape, data_type, 1, fixed_point_position); - beta = create_tensor<TensorType>(param_shape, data_type, 1, fixed_point_position); - gamma = create_tensor<TensorType>(param_shape, data_type, 1, fixed_point_position); + src = create_tensor<TensorType>(tensor_shape, data_type, 1, QuantizationInfo(), data_layout); + dst = create_tensor<TensorType>(tensor_shape, data_type, 1, QuantizationInfo(), data_layout); + mean = create_tensor<TensorType>(param_shape, data_type, 1); + variance = create_tensor<TensorType>(param_shape, data_type, 1); + beta = create_tensor<TensorType>(param_shape, data_type, 1); + gamma = create_tensor<TensorType>(param_shape, data_type, 1); // Create and configure function TensorType *beta_ptr = use_beta ? &beta : nullptr; |