aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/Types.h
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2017-09-26 12:32:57 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:35:24 +0000
commit6f669f039fb74675b858bc3703295609a6a3e122 (patch)
tree704847bbebb2439f68309680bd4f4142b876c179 /arm_compute/graph/Types.h
parent1682430e220eb609752c650f85c0f96e375b6d6a (diff)
downloadComputeLibrary-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/graph/Types.h')
-rw-r--r--arm_compute/graph/Types.h13
1 files changed, 12 insertions, 1 deletions
diff --git a/arm_compute/graph/Types.h b/arm_compute/graph/Types.h
index 0b9596d589..538d64e9bb 100644
--- a/arm_compute/graph/Types.h
+++ b/arm_compute/graph/Types.h
@@ -25,19 +25,24 @@
#define __ARM_COMPUTE_GRAPH_TYPES_H__
#include "arm_compute/core/ITensor.h"
+#include "arm_compute/core/SubTensorInfo.h"
#include "arm_compute/core/TensorInfo.h"
namespace arm_compute
{
namespace graph
{
-using arm_compute::ActivationLayerInfo;
using arm_compute::ITensor;
using arm_compute::TensorInfo;
+using arm_compute::SubTensorInfo;
using arm_compute::DataType;
+using arm_compute::Coordinates;
using arm_compute::TensorShape;
using arm_compute::PadStrideInfo;
using arm_compute::WeightsInfo;
+using arm_compute::ActivationLayerInfo;
+using arm_compute::NormType;
+using arm_compute::NormalizationLayerInfo;
using arm_compute::PoolingLayerInfo;
using arm_compute::PoolingType;
@@ -49,6 +54,12 @@ enum class Hint
NEON /**< Run node on a NEON capable device */
};
+/**< Convolution method hint to the graph executor */
+enum class ConvolutionMethodHint
+{
+ GEMM, /**< Convolution using GEMM */
+ DIRECT /**< Direct convolution */
+};
} // namespace graph
} // namespace arm_compute
#endif /*__ARM_COMPUTE_GRAPH_TYPES_H__*/