From 0f6ca4b83999ef00588fb713e25d5efe98936166 Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Wed, 4 Aug 2021 14:30:28 +0100 Subject: Report error for unsupported non-constant weights in CpuFullyConnected Change-Id: I434844b9194997028e1666f4bce58e69839b886d Signed-off-by: Michele Di Giorgio Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6038 Reviewed-by: Sheri Zhang Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- src/runtime/cpu/operators/CpuFullyConnected.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/src/runtime/cpu/operators/CpuFullyConnected.cpp b/src/runtime/cpu/operators/CpuFullyConnected.cpp index 2b6d051482..e7808fbc82 100644 --- a/src/runtime/cpu/operators/CpuFullyConnected.cpp +++ b/src/runtime/cpu/operators/CpuFullyConnected.cpp @@ -325,6 +325,7 @@ Status CpuFullyConnected::validate(const ITensorInfo *src, const ITensorInfo *we 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(src->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); + ARM_COMPUTE_RETURN_ERROR_ON_MSG(!fc_info.constant_weights, "Non-constant weights are currently not supported"); bool weights_reshaped = fc_info.transpose_weights ? fc_info.are_weights_reshaped : true; bool is_fc_after_conv = true; -- cgit v1.2.1