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,
73 bool enable_fast_math{
false };
74 unsigned int num_groups{ 1 };
88 bool enable_fast_math)
89 : stride(stride), padding(padding), act_info(act_info), dilation(dilation), round_type(round_type), enable_fast_math(enable_fast_math)
98 bool enable_fast_math{
false };
experimental::PostOpList< ITensorInfo * > post_ops
Descriptor used by the 3d Convolution function.
Descriptor used by the FFT1D function.
Descriptor used by the FFT2D function.
Class for specifying the size of a 3D shape or object.
DimensionRoundingType
Dimension rounding type when down-scaling on CNNs.
Conv2dInfo(const PadStrideInfo &conv_info, const Size2D &dilation, const ActivationLayerInfo &act_info, bool enable_fast_math, unsigned int num_groups, const experimental::PostOpList< ITensorInfo *> &post_ops=experimental::PostOpList< ITensorInfo *> {})
Activation Layer Information class.
Copyright (c) 2017-2022 Arm Limited.
const unsigned int num_groups
FFTDirection
FFT direction to use.
Padding and stride information class.
Descriptor used by the 2d Convolution function.
Class for specifying the size of an image or rectangle.
Conv3dInfo(const Size3D &stride, const Padding3D &padding, const ActivationLayerInfo &act_info, const Size3D &dilation, const DimensionRoundingType &round_type, bool enable_fast_math)
A sequence of PostOps that can be appended to the end of other operators.
Padding information for 3D operations like Conv3d.