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 };
108 unsigned int n0{ 0 };
115 bool is_log{
false };
122 int32_t result_fixedpoint_multiplier{ 0 };
123 int32_t result_shift{ 0 };
124 int32_t result_offset_after_shift{ 0 };
143 : gamma(gamma), beta(beta), epsilon(epsilon), use_mixed_precision(use_mixed_precision)
165 : k(k), is_reshaped(is_reshaped), scalar(scalar), mul_by_scalar(mul_by_scalar)
170 bool is_reshaped{
false };
172 bool mul_by_scalar{
false };
191 bool use_padding =
true,
192 bool align_corners =
false,
194 : interpolation_policy{ interpolation_policy },
195 border_mode{ border_mode },
196 constant_border_value{ constant_border_value },
198 use_padding{ use_padding },
199 align_corners{ align_corners },
226 : threshold(threshold), false_value(false_value), true_value(true_value), type(type), upper(upper)
231 uint8_t false_value{ 0 };
232 uint8_t true_value{ 0 };
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.
Descriptor used by the GEMM kernels.
InstanceNormalizationLayerKernelInfo()
Default constructor.
InterpolationPolicy interpolation_policy
Interpolation type to use.
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.
ThresholdKernelInfo(uint8_t threshold, uint8_t false_value, uint8_t true_value, ThresholdType type, uint8_t upper)
Constructor.
const DataLayout data_layout
bool align_corners
Align corners of input and output.
GEMM LHS (Left Hand Side) matrix information.
ThresholdType
Threshold mode.
bool use_padding
Indication of using padding.
Activation Layer Information class.
decltype(strategy::transforms) typedef type
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.
SamplingPolicy sampling_policy
Sampling policy used by the interpolation.
GEMM RHS (Right Hand Side) matrix information.
Descriptor used by the depthwise convolution kernels.
Descriptor used by the FFT core kernels.
Descriptor for FFT scale kernels.
GEMMLowp output stage info.
Descriptor used by the depthwise convolution kernels to retrieve the number of output elements proces...
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.
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)
Constructor.
PixelValue constant_border_value
Constant value to use for constant border mode policy.
float scale
Axis to perform the kernel on.
Threshold with one value.
Descriptor used by the softmax kernels.
DataLayout data_layout
Data layout to use.
DataType
Available data types.
DataLayout
[DataLayout enum definition]
SimpleTensor< T > threshold(const SimpleTensor< T > &src, T threshold, T false_value, T true_value, ThresholdType type, T upper)
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.