diff options
author | Manuel Bottini <manuel.bottini@arm.com> | 2019-07-29 16:59:41 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-07-29 17:11:33 +0000 |
commit | cf3abde6c44177b6db438ecfcdac3d3594d74488 (patch) | |
tree | 406776451b8379bb71513e8006bc15341a84f359 /src/graph/backends/NEON/NEFunctionFactory.cpp | |
parent | c0683b65e2f266fb5b38b767ddc5fe5924eac814 (diff) | |
download | ComputeLibrary-cf3abde6c44177b6db438ecfcdac3d3594d74488.tar.gz |
COMPMID-2430: Adding FusedDepthwiseConvolutionBatchNormalizationLayer condition in NEON function factory
Change-Id: Id7ce542374b1625edbf586ef9688c94ea42f1bb1
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1645
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/graph/backends/NEON/NEFunctionFactory.cpp')
-rw-r--r-- | src/graph/backends/NEON/NEFunctionFactory.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/graph/backends/NEON/NEFunctionFactory.cpp b/src/graph/backends/NEON/NEFunctionFactory.cpp index ad96240a4b..b808ef81f9 100644 --- a/src/graph/backends/NEON/NEFunctionFactory.cpp +++ b/src/graph/backends/NEON/NEFunctionFactory.cpp @@ -223,6 +223,8 @@ std::unique_ptr<IFunction> NEFunctionFactory::create(INode *node, GraphContext & return detail::create_fully_connected_layer<NEFullyConnectedLayer, NETargetInfo>(*polymorphic_downcast<FullyConnectedLayerNode *>(node), ctx); case NodeType::FusedConvolutionBatchNormalizationLayer: return detail::create_fused_convolution_batch_normalization_layer<NEFusedLayerTypes, NETargetInfo>(*polymorphic_downcast<FusedConvolutionBatchNormalizationNode *>(node)); + case NodeType::FusedDepthwiseConvolutionBatchNormalizationLayer: + return detail::create_fused_depthwise_convolution_batch_normalization_layer<NEFusedLayerTypes, NETargetInfo>(*polymorphic_downcast<FusedDepthwiseConvolutionBatchNormalizationNode *>(node)); case NodeType::NormalizationLayer: return detail::create_normalization_layer<NENormalizationLayer, NETargetInfo>(*polymorphic_downcast<NormalizationLayerNode *>(node), ctx); case NodeType::PermuteLayer: |