diff options
author | David Monahan <david.monahan@arm.com> | 2019-06-19 11:47:21 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2019-06-24 23:09:02 +0000 |
commit | 34757810f8b734f5f59485a542b56934ad4cc5f0 (patch) | |
tree | c9859e1eb90fdf1e3ba4fbeadb3952cac3477277 /src/armnnDeserializer/Deserializer.cpp | |
parent | 84da38b0f11ca3db0a439e510514be780f3933ff (diff) | |
download | armnn-34757810f8b734f5f59485a542b56934ad4cc5f0.tar.gz |
IVGCVSW-3235 Add scalar to use as padding value in Reference Pad
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: If050f318fcb7626bbfae1b8737a1d232a4a5a915
Diffstat (limited to 'src/armnnDeserializer/Deserializer.cpp')
-rw-r--r-- | src/armnnDeserializer/Deserializer.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnnDeserializer/Deserializer.cpp b/src/armnnDeserializer/Deserializer.cpp index 452ef8e5ae..5372606689 100644 --- a/src/armnnDeserializer/Deserializer.cpp +++ b/src/armnnDeserializer/Deserializer.cpp @@ -1390,6 +1390,7 @@ void Deserializer::ParsePad(GraphPtr graph, unsigned int layerIndex) auto flatBufferDescriptor = graph->layers()->Get(layerIndex)->layer_as_PadLayer()->descriptor(); auto flatBufferPadList = flatBufferDescriptor->padList(); + float padValue = flatBufferDescriptor->padValue(); if (flatBufferPadList->Length() % 2 != 0) { @@ -1404,7 +1405,7 @@ void Deserializer::ParsePad(GraphPtr graph, unsigned int layerIndex) padList.emplace_back(flatBufferPadList->Get(i), flatBufferPadList->Get(i+1)); } - armnn::PadDescriptor descriptor(padList); + armnn::PadDescriptor descriptor(padList, padValue); auto layerName = GetLayerName(graph, layerIndex); IConnectableLayer* layer = m_Network->AddPadLayer(descriptor, layerName.c_str()); |