23.11
|
Go to the documentation of this file.
24 #ifndef ACL_ARM_COMPUTE_CORE_KERNELDESCRIPTORS_H
25 #define ACL_ARM_COMPUTE_CORE_KERNELDESCRIPTORS_H
64 unsigned int idepth_output_gemm3d,
65 bool ireinterpret_input_as_3d,
67 bool ifp_mixed_precision,
70 int inmult_transpose1xW_width,
71 int imult_interleave4x4_height,
253 #endif // ACL_ARM_COMPUTE_CORE_KERNELDESCRIPTORS_H
Descriptor used by the softmax kernels.
float scale
Axis to perform the kernel on.
bool reinterpret_input_as_3d
Flag used to reinterpret the input as 3D.
unsigned int m
Number of LHS rows.
GEMMRHSMatrixInfo rhs_info
RHS matrix information used for reshaping the RHS matrix.
InstanceNormalizationLayerKernelInfo()
Default constructor.
int32_t a_offset
Offset to be added to each element of the matrix A.
Class describing the value of a pixel for any image format.
DataLayout
[DataLayout enum definition]
GEMMLowp output stage info.
unsigned int n0
Number of columns processed by each thread.
SamplingPolicy
Available Sampling Policies.
int32_t scalar
Scalar value to multiply each reduced column/row by.
unsigned int Nx
Nx coefficient.
bool fp_mixed_precision
Flag used to indicate wider accumulators (32 bit instead of 16 for FP16).
InstanceNormalizationLayerKernelInfo(float gamma, float beta, float epsilon, bool use_mixed_precision)
Constructor.
int32_t result_offset_after_shift
Result offset used for quantizing.
bool adj_rhs
Get Adjoint RHS flag value.
Descriptor for FFT digit reverse kernels.
InterpolationPolicy
Interpolation method.
Descriptor used by the direct convolution layer output stage kernels.
bool is_log
Flag used to perform Log Softmax operation.
bool export_output_to_cl_image
Flag to export the output to cl_image.
unsigned int radix
Radix to use.
Descriptor used by the GEMM kernels.
unsigned int m0
Number of rows processed by each thread.
int m0
Number of output rows processed by each work-item.
Descriptor used by the FFT core kernels.
bool use_padding
Indication of using padding.
int32_t result_shift
Result output stage shift used for quantizing.
DataLayout data_layout
Data layout to use.
Activation Layer Information class.
bool is_first_stage
Flags if the FFT kernels is the first stage of a decomposed FFT.
int k0
Number of inner accumulations.
unsigned int axis
Axis to perform the kernel on.
int32_t n0
Number of columns to be processed by the kernel.
bool use_mixed_precision
Use mixed precision in case of FP16 execution.
BorderMode border_mode
Border mode policy.
unsigned int n
Number of RHS columns.
GEMMLowpReductionKernelInfo()=default
Default constructor.
int32_t result_fixedpoint_multiplier
Result output stage multiplier used for quantizing.
DataType output_data_type
Output tensor data type to use if the output is not initialized.
GEMMLHSMatrixInfo lhs_info
LHS matrix information used to retrieve the number of rows processed by each thread.
Descriptor for FFT scale kernels.
bool align_corners
Align corners of input and output.
bool broadcast_bias
Flag used to broadcast the bias addition.
int32_t axis
The dimension in which to apply softmax.
int32_t m0
Number of rows to be processed by the kernel.
bool export_input_to_cl_image
Export input to cl_image.
bool export_weights_to_cl_image
Export the weights to cl_image.
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.
int mult_transpose1xW_width
Multiplication factor for the width of the 1xW transposed block.
unsigned int k
Number of LHS columns or RHS rows.
float epsilon
Lower bound value for the normalization.
int mult_interleave4x4_height
Multiplication factor for the height of the 4x4 interleaved block.
unsigned int axis
Axis to run the kernel on.
bool has_pad_y
Flag used to indicate if the input/output tensors have internal pad on the y direction.
bool adj_lhs
Get Adjoint LHS flag value.
float gamma
The scale scalar value applied to the normalized tensor.
GEMM LHS (Left Hand Side) matrix information.
GEMMLowpReductionKernelInfo(int32_t k, bool is_reshaped, int32_t scalar, bool mul_by_scalar)
Constructor.
PixelValue constant_border_value
Constant value to use for constant border mode policy.
int32_t k
Number of matrix columns/rows.
GEMMLowpOutputStageInfo output_stage
GEMMLowp output stage information.
MatMulKernelInfo()=default
Copyright (c) 2017-2023 Arm Limited.
Compute descriptor used by the depthwise convolution native kernel.
bool export_weights_to_cl_image
Flag to export the weights to cl_image.
bool export_rhs_to_cl_image
Flag to know whether the RHS tensor should be exported to cl_image.
@ CENTER
Samples are taken at pixel center.
BorderMode
Methods available to handle borders.
ActivationLayerInfo activation_info
Activation function to perform after the matrix multiplication.
bool export_input_to_cl_image
Flag to export the input to cl_image.
bool mul_by_scalar
True if each column/row reduction has to be multiplied by a scalar value.
float beta
A scaling factor for the exponent with default value 1.0.
bool conjugate
Flag to conjugate the output/.
@ UNKNOWN
Unknown data layout.
bool conjugate
Flag to conjugate the output/.
MatMulKernelInfo(bool adj_lhs, bool adj_rhs, int m0=1, int n0=1, int k0=1, bool export_rhs_to_cl_image=false)
unsigned int depth_output_gemm3d
Depth of the output tensor in case is reinterpreted as 3D.
@ UNKNOWN
Unknown data type.
bool is_reshaped
True if the input tensor has been reshaped.
SamplingPolicy sampling_policy
Sampling policy used by the interpolation.
GEMM RHS (Right Hand Side) matrix information.
int n0
Number of output columns processed by each work-item.
float beta
The offset scalar value applied to the normalized tensor.
int32_t k0
Number of partial accumulations to be processed in a single iteration by the kernel.
DataType input_data_type
Input tensor data type.
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)
DataType
Available data types.
Compute descriptor used by the direct convolution kernel.
int32_t b_offset
Offset to be added to each element of the matrix B.