diff options
Diffstat (limited to 'src/armnn/layers/SoftmaxLayer.cpp')
-rw-r--r-- | src/armnn/layers/SoftmaxLayer.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/layers/SoftmaxLayer.cpp b/src/armnn/layers/SoftmaxLayer.cpp index 32d3a1117c..9882da42b0 100644 --- a/src/armnn/layers/SoftmaxLayer.cpp +++ b/src/armnn/layers/SoftmaxLayer.cpp @@ -21,6 +21,8 @@ SoftmaxLayer::SoftmaxLayer(const SoftmaxDescriptor ¶m, const char* name) std::unique_ptr<IWorkload> SoftmaxLayer::CreateWorkload(const IWorkloadFactory& factory) const { SoftmaxQueueDescriptor descriptor; + SetAdditionalInfo(descriptor); + return factory.CreateSoftmax(descriptor, PrepInfoAndDesc(descriptor)); } |