diff options
Diffstat (limited to 'src/armnn/Network.cpp')
-rw-r--r-- | src/armnn/Network.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 7897a81d1e..5c70003785 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -648,9 +648,13 @@ IConnectableLayer* Network::AddDepthwiseConvolution2dLayer( } IConnectableLayer* Network::AddDetectionPostProcessLayer(const armnn::DetectionPostProcessDescriptor& descriptor, - const char* name) + const ConstTensor& anchors, const char* name) { - return m_Graph->AddLayer<DetectionPostProcessLayer>(descriptor, name); + const auto layer = m_Graph->AddLayer<DetectionPostProcessLayer>(descriptor, name); + + layer->m_Anchors = std::make_unique<ScopedCpuTensorHandle>(anchors); + + return layer; } IConnectableLayer* Network::AddPermuteLayer(const PermuteDescriptor& permuteDescriptor, |