diff options
Diffstat (limited to 'src/armnn/backends/NeonWorkloads/NeonMultiplicationFloat32Workload.hpp')
-rw-r--r-- | src/armnn/backends/NeonWorkloads/NeonMultiplicationFloat32Workload.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnn/backends/NeonWorkloads/NeonMultiplicationFloat32Workload.hpp b/src/armnn/backends/NeonWorkloads/NeonMultiplicationFloat32Workload.hpp index ed5ead3700..62e84a2e07 100644 --- a/src/armnn/backends/NeonWorkloads/NeonMultiplicationFloat32Workload.hpp +++ b/src/armnn/backends/NeonWorkloads/NeonMultiplicationFloat32Workload.hpp @@ -9,8 +9,11 @@ namespace armnn { +arm_compute::Status NeonMultiplicationWorkloadValidate(const TensorInfo& input0, + const TensorInfo& input1, + const TensorInfo& output); -class NeonMultiplicationFloat32Workload : public Float32Workload<MultiplicationQueueDescriptor> +class NeonMultiplicationFloat32Workload : public FloatWorkload<MultiplicationQueueDescriptor> { public: NeonMultiplicationFloat32Workload(const MultiplicationQueueDescriptor& descriptor, const WorkloadInfo& info); |