From 39096703269e2ec41fef593a6de49f84ba9c7bb0 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Fri, 20 Nov 2020 04:37:59 +0000 Subject: Validate support of fused activation in NEFullyConnectedLayer Resolves: COMPMID-3989 Signed-off-by: Georgios Pinitas Change-Id: I05bea7f5b0c98b65d3fbb0420117e2723f580c5e Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4495 Reviewed-by: Michele Di Giorgio Reviewed-by: Manuel Bottini Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- src/runtime/NEON/functions/NEFullyConnectedLayer.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/runtime/NEON/functions/NEFullyConnectedLayer.cpp b/src/runtime/NEON/functions/NEFullyConnectedLayer.cpp index 714fa58a66..6b0c27cf65 100644 --- a/src/runtime/NEON/functions/NEFullyConnectedLayer.cpp +++ b/src/runtime/NEON/functions/NEFullyConnectedLayer.cpp @@ -338,6 +338,8 @@ Status NEFullyConnectedLayer::validate(const ITensorInfo *input, const ITensorIn ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(input, weights, output); ARM_COMPUTE_RETURN_ERROR_ON(weights->num_dimensions() > 2); ARM_COMPUTE_RETURN_ERROR_ON(biases != nullptr && biases->num_dimensions() > 1); + ARM_COMPUTE_RETURN_ERROR_ON(fc_info.activation_info.enabled() && is_data_type_quantized(input->data_type()) && fc_info.activation_info.activation() != ActivationLayerInfo::ActivationFunction::RELU + && fc_info.activation_info.activation() != ActivationLayerInfo::ActivationFunction::BOUNDED_RELU && fc_info.activation_info.activation() != ActivationLayerInfo::ActivationFunction::LU_BOUNDED_RELU); bool weights_reshaped = fc_info.transpose_weights ? fc_info.are_weights_reshaped : true; bool is_fc_after_conv = true; -- cgit v1.2.1