24.02.1
|
Go to the documentation of this file.
53 switch (
info.output_data_type)
57 auto k = std::make_unique<kernels::CpuGemmLowpQuantizeDownInt32ToUint8ScaleByFixedPointKernel>();
59 info.gemmlowp_min_bound,
info.gemmlowp_max_bound);
60 _kernel = std::move(k);
65 auto k = std::make_unique<kernels::CpuGemmLowpQuantizeDownInt32ToInt8ScaleByFixedPointKernel>();
67 info.gemmlowp_min_bound,
info.gemmlowp_max_bound);
68 _kernel = std::move(k);
73 auto k = std::make_unique<kernels::CpuGemmLowpQuantizeDownInt32ToInt16ScaleByFixedPointKernel>();
75 info.gemmlowp_max_bound);
76 _kernel = std::move(k);
89 switch (
info.output_data_type)
94 auto k = std::make_unique<kernels::CpuGemmLowpQuantizeDownInt32ScaleKernel>();
96 _kernel = std::move(k);
119 "CpuGemmLowpOutputStage cannot be used with UNKNOWN output data type.");
129 switch (
dst->data_type())
146 switch (
dst->data_type())
static Status validate(const ITensorInfo *src, const ITensorInfo *bias, const ITensorInfo *dst, int min=0, int max=0)
Static function to check if given info will lead to a valid configuration.
SimpleTensor< float > src
static Status validate(const ITensorInfo *src, const ITensorInfo *bias, const ITensorInfo *dst, const GEMMLowpOutputStageInfo *output_stage)
Static function to check if given info will lead to a valid configuration.
GEMMLowp output stage info.
@ QASYMM8
quantized, asymmetric fixed-point 8-bit number unsigned
virtual void schedule_op(ICPPKernel *kernel, const Hints &hints, const Window &window, ITensorPack &tensors)=0
Runs the kernel in the same thread as the caller synchronously.
#define ARM_COMPUTE_ERROR(msg)
Print the given message then throw an std::runtime_error.
#define ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(t, c,...)
@ QSYMM16
quantized, symmetric fixed-point 16-bit number
@ QUANTIZE_DOWN
Quantize using an integer multiplication.
@ RUNTIME_ERROR
Generic runtime error.
static Status validate(const ITensorInfo *src, const ITensorInfo *bias, const ITensorInfo *dst, int min=0, int max=0)
Static function to check if given info will lead to a valid configuration.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
#define ARM_COMPUTE_ERROR_THROW_ON(status)
#define ARM_COMPUTE_RETURN_ERROR_ON(cond)
If the condition is true, an error is returned.
static IScheduler & get()
Access the scheduler singleton.
void configure(ITensorInfo *src, ITensorInfo *bias, ITensorInfo *dst, const GEMMLowpOutputStageInfo &info)
Initialise the kernel's inputs, output.
@ QASYMM8_SIGNED
quantized, asymmetric fixed-point 8-bit number signed
static constexpr size_t DimY
Alias for dimension 1 also known as Y dimension.
static Status validate(const ITensorInfo *src, const ITensorInfo *bias, const ITensorInfo *dst, int min=0, int max=0)
Static function to check if given info will lead to a valid configuration.
#define ARM_COMPUTE_RETURN_ERROR_ON_MSG(cond, msg)
If the condition is true, an error is returned.
Copyright (c) 2017-2024 Arm Limited.
#define ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(...)
Store the tensor's metadata.
#define ARM_COMPUTE_CREATE_ERROR(error_code, msg)
Creates an error with a given message.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
@ UNKNOWN
Unknown data type.
#define ARM_COMPUTE_LOG_PARAMS(...)
@ QUANTIZE_DOWN_FIXEDPOINT
Quantize using a fixed point multiplication.
static Status validate(const ITensorInfo *src, const ITensorInfo *bias, const ITensorInfo *dst, const GEMMLowpOutputStageInfo &info)
Static function to check if given info will lead to a valid configuration.