diff options
Diffstat (limited to 'arm_compute/graph/Types.h')
-rw-r--r-- | arm_compute/graph/Types.h | 13 |
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__*/ |