diff options
| author | Michele Di Giorgio <michele.digiorgio@arm.com> | 2021-08-04 14:30:28 +0100 |
|---|---|---|
| committer | Sheri Zhang <sheri.zhang@arm.com> | 2021-08-04 16:01:59 +0000 |
| commit | 0f6ca4b83999ef00588fb713e25d5efe98936166 (patch) | |
| tree | c26266890fa4cfa5b88becd3fb17af7b0e9ad432 | |
| parent | 6269f87f5caca3a83a4822e0556fbf284158615c (diff) | |
| download | ComputeLibrary-0f6ca4b83999ef00588fb713e25d5efe98936166.tar.gz | |
Report error for unsupported non-constant weights in CpuFullyConnected
Change-Id: I434844b9194997028e1666f4bce58e69839b886d
Signed-off-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6038
Reviewed-by: Sheri Zhang <sheri.zhang@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
| -rw-r--r-- | src/runtime/cpu/operators/CpuFullyConnected.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
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; |