diff options
Diffstat (limited to 'src/armnn/layers/SpaceToBatchNdLayer.cpp')
-rw-r--r-- | src/armnn/layers/SpaceToBatchNdLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/layers/SpaceToBatchNdLayer.cpp b/src/armnn/layers/SpaceToBatchNdLayer.cpp index e801925f58..a4c6d1b237 100644 --- a/src/armnn/layers/SpaceToBatchNdLayer.cpp +++ b/src/armnn/layers/SpaceToBatchNdLayer.cpp @@ -31,7 +31,7 @@ std::unique_ptr<IWorkload> SpaceToBatchNdLayer::CreateWorkload(const IWorkloadFa descriptor.m_Parameters.m_PadList = m_Param.m_PadList; SetAdditionalInfo(descriptor); - return factory.CreateSpaceToBatchNd(descriptor, PrepInfoAndDesc(descriptor)); + return factory.CreateWorkload(LayerType::SpaceToBatchNd, descriptor, PrepInfoAndDesc(descriptor)); } SpaceToBatchNdLayer* SpaceToBatchNdLayer::Clone(Graph& graph) const |