diff options
Diffstat (limited to 'arm_compute/core')
| -rw-r--r-- | arm_compute/core/KernelDescriptors.h | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/arm_compute/core/KernelDescriptors.h b/arm_compute/core/KernelDescriptors.h index fe59365d06..f9f8c141ec 100644 --- a/arm_compute/core/KernelDescriptors.h +++ b/arm_compute/core/KernelDescriptors.h @@ -24,6 +24,8 @@ #ifndef __ARM_COMPUTE_CORE_KERNEL_DESCRIPTORS_H__ #define __ARM_COMPUTE_CORE_KERNEL_DESCRIPTORS_H__ +#include "arm_compute/core/Types.h" + namespace arm_compute { /** Descriptor for FFT scale kernels */ @@ -52,12 +54,13 @@ struct FFTRadixStageKernelInfo /** Descriptor used by the GEMM kernels */ struct GEMMKernelInfo { - unsigned int m{ 0 }; - unsigned int n{ 0 }; - unsigned int k{ 0 }; - unsigned int depth_output_gemm3d{ 0 }; - bool reinterpret_input_as_3d{ false }; - bool broadcast_bias{ false }; + unsigned int m{ 0 }; /**< Number of LHS rows*/ + unsigned int n{ 0 }; /**< Number of RHS columns*/ + unsigned int k{ 0 }; /**< Number of LHS columns or RHS rows */ + unsigned int depth_output_gemm3d{ 0 }; /**< Depth of the output tensor in case is reinterpreted as 3D */ + bool reinterpret_input_as_3d{ false }; /**< Flag used to reinterpret the input as 3D */ + bool broadcast_bias{ false }; /**< Flag used to broadcase the bias addition */ + ActivationLayerInfo activation_info{}; /**< Activation function to perform after the matrix multiplication */ }; } // namespace arm_compute #endif /* __ARM_COMPUTE_CORE_KERNEL_DESCRIPTORS_H__ */ |