From 311cfa7f5fbaad0f98757465390fc9ad20a2d9a8 Mon Sep 17 00:00:00 2001 From: Francis Murtagh Date: Fri, 20 May 2022 13:26:37 +0100 Subject: MLCE-825: Give reason when workload unsupported for Non Constant Weights/Bias * BackendHelper.cpp IsXXXLayerSupported doesn't get as far as Neon/Cl Validate functions where arm_compute::Status is returned. * Conv2d, Depthwise, DilatedDepthwise and FullyConnected * Tidy up if() -> if () * Clean up logic in FullyConnected so that isLayerSupported gets called Signed-off-by: Francis Murtagh Signed-off-by: Teresa Charlin Change-Id: I5da1a882f4a2f55e90aa984b2b9548a847cb3a2d --- src/backends/neon/workloads/NeonFullyConnectedWorkload.cpp | 8 -------- 1 file changed, 8 deletions(-) (limited to 'src/backends/neon/workloads/NeonFullyConnectedWorkload.cpp') diff --git a/src/backends/neon/workloads/NeonFullyConnectedWorkload.cpp b/src/backends/neon/workloads/NeonFullyConnectedWorkload.cpp index d3716806b3..0b91eb37c2 100644 --- a/src/backends/neon/workloads/NeonFullyConnectedWorkload.cpp +++ b/src/backends/neon/workloads/NeonFullyConnectedWorkload.cpp @@ -28,14 +28,6 @@ arm_compute::Status NeonFullyConnectedWorkloadValidate(const TensorInfo& input, const FullyConnectedDescriptor& descriptor, const ActivationDescriptor* activationDescriptor) { - // The NEON implemented workload does support both const and non const - // weights. However, in the case of non const weights we'd have to call - // prepare or configure for each inference which we're not setup to do just yet. - if (!weights.IsConstant()) - { - return arm_compute::Status{arm_compute::ErrorCode::RUNTIME_ERROR, - "Arm NN NeonFullyConnectedWorkload does not support non constant weights."}; - } const arm_compute::TensorInfo aclInput = BuildArmComputeTensorInfo(input); const arm_compute::TensorInfo aclOutput = BuildArmComputeTensorInfo(output); arm_compute::TensorInfo aclWeights = BuildArmComputeTensorInfo(weights); -- cgit v1.2.1