24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_GPU_CL_OPERATORS_CLSOFTMAX_H
25 #define ACL_SRC_GPU_CL_OPERATORS_CLSOFTMAX_H
34 class CLCompileContext;
37 struct SoftmaxKernelInfo;
43 class ClSoftmaxKernel;
74 enum InternalTensorIdx
86 #endif // ACL_SRC_GPU_CL_OPERATORS_CLSOFTMAX_H
Descriptor used by the softmax kernels.
std::vector< MemoryInfo > MemoryRequirements
SimpleTensor< float > src
static Status validate(const ITensorInfo &src, const ITensorInfo &dst, const SoftmaxKernelInfo &info)
Static function to check if given info will lead to a valid configuration.
void configure(const CLCompileContext &compile_context, const ITensorInfo &src, ITensorInfo &dst, const SoftmaxKernelInfo &info)
Configure the operator.
Basic interface for functions which have a single async CL kernel.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
Store the tensor's metadata.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)