diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-11-20 04:37:59 +0000 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-11-20 12:00:57 +0000 |
| commit | 39096703269e2ec41fef593a6de49f84ba9c7bb0 (patch) | |
| tree | 37af4c0b77a86352e4abf7f7fe6d98384acd95a6 | |
| parent | 16846ab2627074cafa8abb698fc6762bdaed0703 (diff) | |
| download | ComputeLibrary-39096703269e2ec41fef593a6de49f84ba9c7bb0.tar.gz | |
Validate support of fused activation in NEFullyConnectedLayer
Resolves: COMPMID-3989
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I05bea7f5b0c98b65d3fbb0420117e2723f580c5e
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4495
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: Manuel Bottini <manuel.bottini@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
| -rw-r--r-- | src/runtime/NEON/functions/NEFullyConnectedLayer.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
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; |