diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-09-26 12:32:57 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
| commit | 6f669f039fb74675b858bc3703295609a6a3e122 (patch) | |
| tree | 704847bbebb2439f68309680bd4f4142b876c179 /arm_compute/core/SubTensorInfo.h | |
| parent | 1682430e220eb609752c650f85c0f96e375b6d6a (diff) | |
| download | ComputeLibrary-6f669f039fb74675b858bc3703295609a6a3e122.tar.gz | |
COMPMID-417: Add grouping in convolution layer
-Adds grouping support in convolution layer
-Adds Normalization layer node in graph
-Adds alexnet example
-Fixes FullyConnectedLayer output autoconfigure (works only for 1d batch
space)
Change-Id: I5bd75f9a8b08cfd68f7c34745150266c2bc4221f
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/89518
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/core/SubTensorInfo.h')
| -rw-r--r-- | arm_compute/core/SubTensorInfo.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/arm_compute/core/SubTensorInfo.h b/arm_compute/core/SubTensorInfo.h index e2532fd487..54fb66a573 100644 --- a/arm_compute/core/SubTensorInfo.h +++ b/arm_compute/core/SubTensorInfo.h @@ -27,6 +27,7 @@ #include "arm_compute/core/ITensorInfo.h" #include "arm_compute/core/Coordinates.h" +#include "arm_compute/core/Helpers.h" #include "arm_compute/core/Strides.h" #include "arm_compute/core/TensorInfo.h" #include "arm_compute/core/TensorShape.h" |