24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_GPU_CL_KERNELS_CLSOFTMAXKERNEL_H
25 #define ACL_SRC_GPU_CL_KERNELS_CLSOFTMAXKERNEL_H
77 bool _prepared{
false};
85 #endif // ACL_SRC_GPU_CL_KERNELS_CLSOFTMAXKERNEL_H
Descriptor used by the softmax kernels.
SimpleTensor< float > src
void run_op(ITensorPack &tensors, const Window &window, cl::CommandQueue &queue) override
Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(ClSoftmaxKernel)
static Status validate(const ITensorInfo &src, const ITensorInfo &dst, const SoftmaxKernelInfo &info)
Check if the kernel arguments are valid.
The CL kernel that performs softmax function.
const TensorInfo & tmp_tensor_info() const
Get the tensor info of the temporary tensor.
Common interface for all the OpenCL kernels.
const Window & window() const
The maximum window the kernel can be executed on.
Store the tensor's metadata.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
void configure(const CLCompileContext &compile_context, const ITensorInfo &src, ITensorInfo &dst, const SoftmaxKernelInfo &info)
Configure the kernel.