aboutsummaryrefslogtreecommitdiff
path: root/src/armnnDeserializer/Deserializer.hpp
diff options
context:
space:
mode:
authorNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2019-02-26 15:53:02 +0000
committerderek.lamberti <derek.lamberti@arm.com>2019-02-26 16:49:44 +0000
commit452869973b9a45c9c44820d16f92f7dfc96e9aef (patch)
treefe4c176852db25435da8865008c626cbd064d7bf /src/armnnDeserializer/Deserializer.hpp
parent07d58c77e174a8e041c1facc61637e85c6cc0bf6 (diff)
downloadarmnn-452869973b9a45c9c44820d16f92f7dfc96e9aef.tar.gz
IVGCVSW-2708 Add Serializer and Deserializer SpaceToBatchNd
Change-Id: I14f2a660654dc87939fe413fafa9384f09594f68 Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'src/armnnDeserializer/Deserializer.hpp')
-rw-r--r--src/armnnDeserializer/Deserializer.hpp1
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnnDeserializer/Deserializer.hpp b/src/armnnDeserializer/Deserializer.hpp
index 9159ff2f36..c88459d97a 100644
--- a/src/armnnDeserializer/Deserializer.hpp
+++ b/src/armnnDeserializer/Deserializer.hpp
@@ -79,6 +79,7 @@ private:
void ParsePooling2d(GraphPtr graph, unsigned int layerIndex);
void ParseReshape(GraphPtr graph, unsigned int layerIndex);
void ParseSoftmax(GraphPtr graph, unsigned int layerIndex);
+ void ParseSpaceToBatchNd(GraphPtr graph, unsigned int layerIndex);
void RegisterOutputSlotOfConnection(uint32_t connectionIndex, armnn::IOutputSlot* slot);
void RegisterInputSlotOfConnection(uint32_t connectionIndex, armnn::IInputSlot* slot);