24 #ifndef ARM_COMPUTE_RUNTIME_FUNCTION_DESCRIPTORS_H 25 #define ARM_COMPUTE_RUNTIME_FUNCTION_DESCRIPTORS_H 43 unsigned int axis{ 0 };
50 unsigned int axis0{ 0 };
51 unsigned int axis1{ 1 };
63 bool enable_fast_math,
65 : conv_info(conv_info), dilation(dilation), act_info(act_info), enable_fast_math(enable_fast_math), num_groups(num_groups)
72 bool enable_fast_math{
false };
Descriptor used by the FFT1D function.
Descriptor used by the FFT2D function.
Activation Layer Information class.
Copyright (c) 2017-2021 Arm Limited.
Conv2dInfo(const PadStrideInfo &conv_info, const Size2D &dilation, const ActivationLayerInfo &act_info, bool enable_fast_math, unsigned int num_groups)
const unsigned int num_groups
FFTDirection
FFT direction to use.
Padding and stride information class.
Descriptor used by the Convolution function.
Class for specifying the size of an image or rectangle.