24 #ifndef ARM_COMPUTE_CORE_KERNEL_DESCRIPTORS_H 25 #define ARM_COMPUTE_CORE_KERNEL_DESCRIPTORS_H 43 unsigned int axis{ 0 };
50 unsigned int axis{ 0 };
51 unsigned int radix{ 0 };
53 bool is_first_stage{
false };
65 unsigned int idepth_output_gemm3d,
66 bool ireinterpret_input_as_3d,
68 bool ifp_mixed_precision,
71 int inmult_transpose1xW_width,
72 int imult_interleave4x4_height,
78 :
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),
79 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),
80 rhs_info(irhs_info), a_offset(ina_offset), b_offset(inb_offset),
post_ops(ipost_ops)
87 unsigned int depth_output_gemm3d{ 0 };
88 bool reinterpret_input_as_3d{
false };
90 bool fp_mixed_precision{
false };
91 bool has_pad_y{
false };
93 int mult_transpose1xW_width{ 1 };
94 int mult_interleave4x4_height{ 1 };
97 int32_t a_offset{ 0 };
98 int32_t b_offset{ 0 };
107 unsigned int n0{ 0 };
108 unsigned int m0{ 0 };
116 bool is_log{
false };
124 int32_t result_fixedpoint_multiplier{ 0 };
125 int32_t result_shift{ 0 };
126 int32_t result_offset_after_shift{ 0 };
145 : gamma(gamma), beta(beta), epsilon(epsilon), use_mixed_precision(use_mixed_precision)
167 : k(k), is_reshaped(is_reshaped), scalar(scalar), mul_by_scalar(mul_by_scalar)
172 bool is_reshaped{
false };
174 bool mul_by_scalar{
false };
193 bool use_padding =
true,
194 bool align_corners =
false,
196 : interpolation_policy{ interpolation_policy },
197 border_mode{ border_mode },
198 constant_border_value{ constant_border_value },
200 use_padding{ use_padding },
201 align_corners{ align_corners },
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.
experimental::PostOpList< ITensorInfo * > post_ops
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.
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, const experimental::PostOpList< ITensorInfo *> &ipost_ops=experimental::PostOpList< ITensorInfo *> {})
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.
Store the tensor's metadata.
bool use_padding
Indication of using padding.
Activation Layer Information class.
Copyright (c) 2017-2022 Arm Limited.
Samples are taken at pixel center.
bool use_mixed_precision
Use mixed precision in case of FP16 execution.
const OutputStage & output_stage
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 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.
DataLayout
[DataLayout enum definition]
A sequence of PostOps that can be appended to the end of other operators.
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.