diff options
Diffstat (limited to 'tests/benchmark/fixtures/FullyConnectedLayerFixture.h')
-rw-r--r-- | tests/benchmark/fixtures/FullyConnectedLayerFixture.h | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/tests/benchmark/fixtures/FullyConnectedLayerFixture.h b/tests/benchmark/fixtures/FullyConnectedLayerFixture.h index e7a5260f44..caef5bebc9 100644 --- a/tests/benchmark/fixtures/FullyConnectedLayerFixture.h +++ b/tests/benchmark/fixtures/FullyConnectedLayerFixture.h @@ -45,15 +45,15 @@ public: void setup(TensorShape src_shape, TensorShape weights_shape, TensorShape biases_shape, TensorShape dst_shape, DataType data_type, int batches) { // Set batched in source and destination shapes - const unsigned int fixed_point_position = 4; + src_shape.set(src_shape.num_dimensions() /* batch */, batches); dst_shape.set(dst_shape.num_dimensions() /* batch */, batches); // Create tensors - src = create_tensor<TensorType>(src_shape, data_type, 1, fixed_point_position); - weights = create_tensor<TensorType>(weights_shape, data_type, 1, fixed_point_position); - biases = create_tensor<TensorType>(biases_shape, data_type, 1, fixed_point_position); - dst = create_tensor<TensorType>(dst_shape, data_type, 1, fixed_point_position); + src = create_tensor<TensorType>(src_shape, data_type, 1); + weights = create_tensor<TensorType>(weights_shape, data_type, 1); + biases = create_tensor<TensorType>(biases_shape, data_type, 1); + dst = create_tensor<TensorType>(dst_shape, data_type, 1); // Create and configure function fc_layer.configure(&src, &weights, &biases, &dst); |