24 #ifndef ARM_COMPUTE_CLSOFTMAXLAYER_H 25 #define ARM_COMPUTE_CLSOFTMAXLAYER_H 37 class CLCompileContext;
38 class CLLogits1DMaxShiftExpSumKernel;
39 class CLLogits1DNormKernel;
57 template <
bool IS_LOG = false>
111 std::unique_ptr<CLLogits1DMaxShiftExpSumKernel> _max_shift_exp_sum_kernel;
112 std::unique_ptr<CLLogits1DNormKernel> _norm_kernel;
void run() override
Run the kernels contained in the function.
Base class for all functions.
CLSoftmaxLayerGeneric(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Constructor.
void configure(const ICLTensor *input, ICLTensor *output, float beta=1.0f, int32_t axis=0)
Set the input and output tensors.
Store the tensor's metadata.
Copyright (c) 2017-2021 Arm Limited.
Basic function to compute a SoftmaxLayer.
Basic function to execute an opencl::kernels::ClPermuteKernel.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, float beta=1.0f, int32_t axis=0)
Static function to check if given info will lead to a valid configuration of CLSoftmaxLayer.
Interface for OpenCL tensor.
CLSoftmaxLayerGeneric & operator=(const CLSoftmaxLayerGeneric &)=delete
Prevent instances of this class from being copied.
~CLSoftmaxLayerGeneric()
Default destructor.
Basic implementation of the OpenCL tensor interface.