diff options
Diffstat (limited to 'src/armnn/Network.cpp')
-rw-r--r-- | src/armnn/Network.cpp | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 63432da0ff..f9115ea5df 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -1182,16 +1182,22 @@ IConnectableLayer* Network::AddBatchNormalizationLayer(const BatchNormalizationD return layer; } -IConnectableLayer* Network::AddResizeBilinearLayer(const ResizeBilinearDescriptor& -resizeDescriptor, const char* name) +IConnectableLayer* Network::AddResizeBilinearLayer(const ResizeBilinearDescriptor& descriptor, + const char* name) { - return m_Graph->AddLayer<ResizeBilinearLayer>(resizeDescriptor,name); + ResizeDescriptor resizeDescriptor; + resizeDescriptor.m_Method = ResizeMethod::Bilinear; + resizeDescriptor.m_DataLayout = descriptor.m_DataLayout; + resizeDescriptor.m_TargetWidth = descriptor.m_TargetWidth; + resizeDescriptor.m_TargetHeight = descriptor.m_TargetHeight; + + return m_Graph->AddLayer<ResizeLayer>(resizeDescriptor, name); } IConnectableLayer* Network::AddResizeLayer(const ResizeDescriptor& resizeDescriptor, const char* name) { - return m_Graph->AddLayer<ResizeLayer>(resizeDescriptor,name); + return m_Graph->AddLayer<ResizeLayer>(resizeDescriptor, name); } IConnectableLayer* Network::AddL2NormalizationLayer(const L2NormalizationDescriptor& desc, |