diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-11-02 01:37:17 +0000 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-11-12 15:59:25 +0000 |
| commit | c0b6f76561580414f08633a804fc548ccad65659 (patch) | |
| tree | 4d46b7f479de04f799e29095392948aeb370c029 /arm_compute/runtime/FunctionDescriptors.h | |
| parent | 824061d9910ebb42cbe46b677c0b843db212c9a2 (diff) | |
| download | ComputeLibrary-c0b6f76561580414f08633a804fc548ccad65659.tar.gz | |
COMPMID-3776: Indirect GEMM
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I51a1b0f098bc3a8c408c50c92221e4df3061e12c
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4343
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/FunctionDescriptors.h')
| -rw-r--r-- | arm_compute/runtime/FunctionDescriptors.h | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/arm_compute/runtime/FunctionDescriptors.h b/arm_compute/runtime/FunctionDescriptors.h index 16d6c345e2..1f4216eb21 100644 --- a/arm_compute/runtime/FunctionDescriptors.h +++ b/arm_compute/runtime/FunctionDescriptors.h @@ -23,6 +23,9 @@ */ #ifndef ARM_COMPUTE_RUNTIME_FUNCTION_DESCRIPTORS_H #define ARM_COMPUTE_RUNTIME_FUNCTION_DESCRIPTORS_H + +#include "arm_compute/core/Types.h" + #include <utility> namespace arm_compute @@ -48,5 +51,26 @@ struct FFT2DInfo unsigned int axis1{ 1 }; /**< Axis to run second pass on. If same, multiple transforms are performed on single axis*/ FFTDirection direction{ FFTDirection::Forward }; /**< Direction of the FFT. */ }; + +/** Descriptor used by the Convolution function */ +struct Conv2dInfo +{ + Conv2dInfo() = default; + + Conv2dInfo(const PadStrideInfo &conv_info, + const Size2D &dilation, + const ActivationLayerInfo &act_info, + bool enable_fast_math, + unsigned int num_groups) + : conv_info(conv_info), dilation(dilation), act_info(act_info), enable_fast_math(enable_fast_math), num_groups(num_groups) + { + } + + PadStrideInfo conv_info{}; + Size2D dilation{ 1U, 1U }; + ActivationLayerInfo act_info{}; + bool enable_fast_math{ false }; + unsigned int num_groups{ 1 }; +}; } // namespace arm_compute #endif /* ARM_COMPUTE_RUNTIME_FUNCTION_DESCRIPTORS_H */ |