24 #ifndef ARM_COMPUTE_CLSOFTMAXLAYER_H 25 #define ARM_COMPUTE_CLSOFTMAXLAYER_H 37 class CLCompileContext;
53 template <
bool IS_LOG = false>
108 std::unique_ptr<Impl> _impl;
111 void allocate_workspace();
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.
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()
Default destructor.