24 #ifndef ARM_COMPUTE_CORE_KERNEL_DESCRIPTORS_H 25 #define ARM_COMPUTE_CORE_KERNEL_DESCRIPTORS_H 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,
108 unsigned int n0{ 0 };
BorderMode
Methods available to handle borders.
int32_t result_fixedpoint_multiplier
Result output stage multiplier used for quantizing.
BorderMode border_mode
Border mode policy.
Class describing the value of a pixel for any image format.
bool broadcast_bias
Flag used to broadcast the bias addition.
InterpolationPolicy
Interpolation method.
int mult_interleave4x4_height
Multiplication factor for the height of the 4x4 interleaved block.
bool fp_mixed_precision
Flag used to indicate wider accumulators (32 bit instead of 16 for FP16).
Descriptor used by the GEMM kernels.
InstanceNormalizationLayerKernelInfo()
Default constructor.
InterpolationPolicy interpolation_policy
Interpolation type to use.
int32_t result_offset_after_shift
Result offset used for quantizing.
unsigned int n0
Number of columns processed by each thread.
float beta
A scaling factor for the exponent with default value 1.0.
unsigned int depth_output_gemm3d
Depth of the output tensor in case is reinterpreted as 3D.
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.
bool align_corners
Align corners of input and output.
GEMM LHS (Left Hand Side) matrix information.
ActivationLayerInfo activation_info
Activation function to perform after the depthwise convolution.
bool use_padding
Indication of using padding.
unsigned int axis
Axis to run the kernel on.
int32_t scalar
Scalar value to multiply each reduced column/row by.
ActivationLayerInfo activation_info
Activation function to perform after the matrix multiplication.
Activation Layer Information class.
GEMMLHSMatrixInfo lhs_info
LHS matrix information used to retrieve the number of rows processed by each thread.
Copyright (c) 2017-2021 Arm Limited.
Samples are taken at pixel center.
bool use_mixed_precision
Use mixed precision in case of FP16 execution.
unsigned int m
Number of LHS rows.
unsigned int n
Number of RHS columns.
GEMMLowpReductionKernelInfo(int32_t k, bool is_reshaped, int32_t scalar, bool mul_by_scalar)
Constructor.
int32_t k
Number of matrix columns/rows.
SamplingPolicy sampling_policy
Sampling policy used by the interpolation.
GEMM RHS (Right Hand Side) matrix information.
int32_t b_offset
Offset to be added to each element of the matrix B.
bool conjugate
Flag to conjugate the output/.
DataType input_data_type
Input tensor data type.
bool is_log
Flag used to perform Log Softmax operation.
Descriptor used by the depthwise convolution kernels.
GEMMLowpOutputStageInfo output_stage
GEMMLowp output stage information.
Descriptor used by the FFT core kernels.
bool reinterpret_input_as_3d
Flag used to reinterpret the input as 3D.
Descriptor for FFT scale kernels.
GEMMLowp output stage info.
Descriptor used by the depthwise convolution kernels to retrieve the number of output elements proces...
unsigned int axis
Axis to perform the kernel on.
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.
bool has_pad_y
Flag used to indicate if the input/output tensors have internal pad on the y direction.
int32_t a_offset
Offset to be added to each element of the matrix A.
float epsilon
Lower bound value for the normalization.
unsigned int radix
Radix to use.
GEMMRHSMatrixInfo rhs_info
RHS matrix information used for reshaping the RHS matrix.
bool is_first_stage
Flags if the FFT kernels is the first stage of a decomposed FFT.
Descriptor for FFT digit reverse kernels.
unsigned int Nx
Nx coefficient.
float gamma
The scale scalar value applied to the normalized tensor.
DataType output_data_type
Output tensor data type to use if the output is not initialized.
GEMMLowpReductionKernelInfo()=default
Default constructor.
unsigned int k
Number of LHS columns or RHS rows.
bool is_reshaped
True if the input tensor has been reshaped.
PixelValue constant_border_value
Constant value to use for constant border mode policy.
int32_t axis
The dimension in which to apply softmax.
float scale
Axis to perform the kernel on.
Descriptor used by the softmax kernels.
int mult_transpose1xW_width
Multiplication factor for the width of the 1xW transposed block.
DataLayout data_layout
Data layout to use.
DataType
Available data types.
bool mul_by_scalar
True if each column/row reduction has to be multiplied by a scalar 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.
int32_t result_shift
Result output stage shift used for quantizing.