diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/benchmark/fixtures/WinogradConvolutionLayerFixture.h | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tests/benchmark/fixtures/WinogradConvolutionLayerFixture.h b/tests/benchmark/fixtures/WinogradConvolutionLayerFixture.h index 5f44517817..86ea9c4dc1 100644 --- a/tests/benchmark/fixtures/WinogradConvolutionLayerFixture.h +++ b/tests/benchmark/fixtures/WinogradConvolutionLayerFixture.h @@ -60,7 +60,11 @@ public: dst = create_tensor<TensorType>(dst_shape, data_type, 1); // Create and configure function - conv_layer.configure(&src, &weights, &biases, &dst, info, act_info); + if(data_type == DataType::F16){ + conv_layer.configure(&src, &weights, &biases, &dst, info, act_info, true); + }else{ + conv_layer.configure(&src, &weights, &biases, &dst, info, act_info); + } // Allocate tensors src.allocator()->allocate(); |