diff options
Diffstat (limited to 'src/backends/neon/NeonWorkloadFactory.cpp')
-rw-r--r-- | src/backends/neon/NeonWorkloadFactory.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/backends/neon/NeonWorkloadFactory.cpp b/src/backends/neon/NeonWorkloadFactory.cpp index cbed690733..f087c6262a 100644 --- a/src/backends/neon/NeonWorkloadFactory.cpp +++ b/src/backends/neon/NeonWorkloadFactory.cpp @@ -320,6 +320,8 @@ std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateWorkload(LayerType type, } case UnaryOperation::Sin: return std::make_unique<NeonSinWorkload>(*elementwiseUnaryQueueDescriptor, info); + case UnaryOperation::Sqrt: + return std::make_unique<NeonSqrtWorkload>(*elementwiseUnaryQueueDescriptor, info); default: return nullptr; } |