COMPMID-1386: Add support for converting weights for CL.
Change-Id: I62e3ead903366baeeb1488f233a9b8b0c388c9de
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/140403
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
diff --git a/tests/validation/CL/FullyConnectedLayer.cpp b/tests/validation/CL/FullyConnectedLayer.cpp
index 9958a88..cd050e3 100644
--- a/tests/validation/CL/FullyConnectedLayer.cpp
+++ b/tests/validation/CL/FullyConnectedLayer.cpp
@@ -96,9 +96,14 @@
const QuantizationInfo src_quantization_info = src.info()->quantization_info();
const QuantizationInfo weights_quantization_info = weights.info()->quantization_info();
+ // 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.
CLFullyConnectedLayer 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);
@@ -141,12 +146,16 @@
framework::dataset::make("Expected", { false, true, true, false, false })),
input_info, weights_info, bias_info, output_info, transpose_weights, reshaped_weights, expected)
{
+ // Create Fully Connected layer info
+ FullyConnectedLayerInfo fc_info;
+ fc_info.transpose_weights = transpose_weights;
+ fc_info.are_weights_reshaped = reshaped_weights;
+
Status status = CLFullyConnectedLayer::validate(&input_info.clone()->set_is_resizable(false),
&weights_info.clone()->set_is_resizable(false),
&bias_info.clone()->set_is_resizable(false),
&output_info.clone()->set_is_resizable(false),
- transpose_weights,
- reshaped_weights);
+ fc_info);
ARM_COMPUTE_EXPECT(bool(status) == expected, framework::LogLevel::ERRORS);
}
// clang-format on