24 #ifndef ARM_COMPUTE_CORE_KERNEL_DESCRIPTORS_H 25 #define ARM_COMPUTE_CORE_KERNEL_DESCRIPTORS_H 42 unsigned int axis{ 0 };
49 unsigned int axis{ 0 };
50 unsigned int radix{ 0 };
52 bool is_first_stage{
false };
63 unsigned int idepth_output_gemm3d,
64 bool ireinterpret_input_as_3d,
66 bool ifp_mixed_precision,
69 int inmult_transpose1xW_width,
70 int imult_interleave4x4_height,
75 : m(im), n(in), k(ik), depth_output_gemm3d(idepth_output_gemm3d), reinterpret_input_as_3d(ireinterpret_input_as_3d), broadcast_bias(ibroadcast_bias), fp_mixed_precision(ifp_mixed_precision),
76 has_pad_y(ihas_pad_y), activation_info(iactivation_info), mult_transpose1xW_width(inmult_transpose1xW_width), mult_interleave4x4_height(imult_interleave4x4_height), lhs_info(ilhs_info),
77 rhs_info(irhs_info), a_offset(ina_offset), b_offset(inb_offset)
84 unsigned int depth_output_gemm3d{ 0 };
85 bool reinterpret_input_as_3d{
false };
86 bool broadcast_bias{
false };
87 bool fp_mixed_precision{
false };
88 bool has_pad_y{
false };
90 int mult_transpose1xW_width{ 1 };
91 int mult_interleave4x4_height{ 1 };
94 int32_t a_offset{ 0 };
95 int32_t b_offset{ 0 };
102 unsigned int n0{ 0 };
103 unsigned int m0{ 0 };
111 bool is_log{
false };
119 int32_t result_fixedpoint_multiplier{ 0 };
120 int32_t result_shift{ 0 };
121 int32_t result_offset_after_shift{ 0 };
140 : gamma(gamma), beta(beta), epsilon(epsilon), use_mixed_precision(use_mixed_precision)
162 : k(k), is_reshaped(is_reshaped), scalar(scalar), mul_by_scalar(mul_by_scalar)
167 bool is_reshaped{
false };
169 bool mul_by_scalar{
false };
188 bool use_padding =
true,
189 bool align_corners =
false,
191 : interpolation_policy{ interpolation_policy },
192 border_mode{ border_mode },
193 constant_border_value{ constant_border_value },
195 use_padding{ use_padding },
196 align_corners{ align_corners },
214 : policy(policy), border_mode(border_mode), constant_border_value(constant_border_value)
BorderMode
Methods available to handle borders.
BorderMode border_mode
Border mode policy.
Class describing the value of a pixel for any image format.
InterpolationPolicy
Interpolation method.
bool export_weights_to_cl_image(const ITensorInfo *tensor)
Descriptor used by the GEMM kernels.
InstanceNormalizationLayerKernelInfo()
Default constructor.
InterpolationPolicy interpolation_policy
Interpolation type to use.
PixelValue constant_border_value
GEMMKernelInfo(unsigned int im, unsigned int in, unsigned int ik, unsigned int idepth_output_gemm3d, bool ireinterpret_input_as_3d, bool ibroadcast_bias, bool ifp_mixed_precision, bool ihas_pad_y, ActivationLayerInfo iactivation_info, int inmult_transpose1xW_width, int imult_interleave4x4_height, GEMMLHSMatrixInfo ilhs_info, GEMMRHSMatrixInfo irhs_info, int32_t ina_offset, int32_t inb_offset)
InstanceNormalizationLayerKernelInfo(float gamma, float beta, float epsilon, bool use_mixed_precision)
Constructor.
const DataLayout data_layout
bool align_corners
Align corners of input and output.
GEMM LHS (Left Hand Side) matrix information.
bool use_padding
Indication of using padding.
Activation Layer Information class.
Copyright (c) 2017-2021 Arm Limited.
Samples are taken at pixel center.
bool use_mixed_precision
Use mixed precision in case of FP16 execution.
GEMMLowpReductionKernelInfo(int32_t k, bool is_reshaped, int32_t scalar, bool mul_by_scalar)
Constructor.
InterpolationPolicy policy
SamplingPolicy sampling_policy
Sampling policy used by the interpolation.
GEMM RHS (Right Hand Side) matrix information.
Descriptor used by the FFT core kernels.
Descriptor for FFT scale kernels.
GEMMLowp output stage info.
Compute descriptor used by the depthwise convolution native kernel.
ScaleKernelInfo(InterpolationPolicy interpolation_policy, BorderMode border_mode, PixelValue constant_border_value=PixelValue(), SamplingPolicy sampling_policy=SamplingPolicy::CENTER, bool use_padding=true, bool align_corners=false, DataLayout data_layout=DataLayout::UNKNOWN) noexcept
Constructor.
float epsilon
Lower bound value for the normalization.
Descriptor for FFT digit reverse kernels.
float gamma
The scale scalar value applied to the normalized tensor.
PixelValue constant_border_value
Constant value to use for constant border mode policy.
float scale
Axis to perform the kernel on.
Descriptor used by the softmax kernels.
DataLayout data_layout
Data layout to use.
DataType
Available data types.
RemapInfo(InterpolationPolicy policy, BorderMode border_mode, PixelValue constant_border_value)
DataLayout
[DataLayout enum definition]
float beta
The offset scalar value applied to the normalized tensor.
Descriptor used by the direct convolution layer output stage kernels.
bool conjugate
Flag to conjugate the output/.
SamplingPolicy
Available Sampling Policies.