diff options
Diffstat (limited to 'tests/validation/GLES_COMPUTE/FullyConnectedLayer.cpp')
-rw-r--r-- | tests/validation/GLES_COMPUTE/FullyConnectedLayer.cpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/tests/validation/GLES_COMPUTE/FullyConnectedLayer.cpp b/tests/validation/GLES_COMPUTE/FullyConnectedLayer.cpp index 49716dc946..c82a8a1a43 100644 --- a/tests/validation/GLES_COMPUTE/FullyConnectedLayer.cpp +++ b/tests/validation/GLES_COMPUTE/FullyConnectedLayer.cpp @@ -87,9 +87,14 @@ DATA_TEST_CASE(Configuration, framework::DatasetMode::ALL, combine(combine(frame ARM_COMPUTE_EXPECT(bias.info()->is_resizable(), framework::LogLevel::ERRORS); ARM_COMPUTE_EXPECT(dst.info()->is_resizable(), framework::LogLevel::ERRORS); + // Create Fully Connected layer info + FullyConnectedLayerInfo fc_info; + fc_info.transpose_weights = transpose_weights; + fc_info.are_weights_reshaped = !reshape_weights; + // Create and configure function. GCFullyConnectedLayer fc; - fc.configure(&src, &weights, &bias, &dst, transpose_weights, !reshape_weights); + fc.configure(&src, &weights, &bias, &dst, fc_info); // Validate valid region const ValidRegion dst_valid_region = shape_to_valid_region(dst_shape); |