Interface for calculating the final step of the Softmax Layer where each logit value is multiplied by the inverse of the sum of the logits.
More...
|
| CLLogits1DNormKernel () |
| Default constructor. More...
|
|
| CLLogits1DNormKernel (const CLLogits1DNormKernel &)=delete |
| Prevent instances of this class from being copied (As this class contains pointers) More...
|
|
CLLogits1DNormKernel & | operator= (const CLLogits1DNormKernel &)=delete |
| Prevent instances of this class from being copied (As this class contains pointers) More...
|
|
| CLLogits1DNormKernel (CLLogits1DNormKernel &&)=default |
| Allow instances of this class to be moved. More...
|
|
CLLogits1DNormKernel & | operator= (CLLogits1DNormKernel &&)=default |
| Allow instances of this class to be moved. More...
|
|
void | configure (const ICLTensor *input, const ICLTensor *sum, ICLTensor *output) |
| Set the input and output tensors. More...
|
|
void | run (const Window &window, cl::CommandQueue &queue) override |
| Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue. More...
|
|
| ICLKernel () |
| Constructor. More...
|
|
cl::Kernel & | kernel () |
| Returns a reference to the OpenCL kernel of this object. More...
|
|
void | add_1D_tensor_argument (unsigned int &idx, const ICLTensor *tensor, const Window &window) |
| Add the passed 1D tensor's parameters to the object's kernel's arguments starting from the index idx. More...
|
|
void | add_2D_tensor_argument (unsigned int &idx, const ICLTensor *tensor, const Window &window) |
| Add the passed 2D tensor's parameters to the object's kernel's arguments starting from the index idx. More...
|
|
void | add_3D_tensor_argument (unsigned int &idx, const ICLTensor *tensor, const Window &window) |
| Add the passed 3D tensor's parameters to the object's kernel's arguments starting from the index idx. More...
|
|
unsigned int | num_arguments_per_1D_tensor () const |
| Returns the number of arguments enqueued per 1D tensor object. More...
|
|
unsigned int | num_arguments_per_2D_tensor () const |
| Returns the number of arguments enqueued per 2D tensor object. More...
|
|
unsigned int | num_arguments_per_3D_tensor () const |
| Returns the number of arguments enqueued per 3D tensor object. More...
|
|
template<typename T > |
void | add_argument (unsigned int &idx, T value) |
| Add the passed parameters to the object's kernel's arguments starting from the index idx. More...
|
|
| IKernel () |
| Constructor. More...
|
|
virtual | ~IKernel ()=default |
| Destructor. More...
|
|
virtual bool | is_parallelisable () const |
| Indicates whether or not the kernel is parallelisable. More...
|
|
virtual BorderSize | border_size () const |
| The size of the border for that kernel. More...
|
|
const Window & | window () const |
| The maximum window the kernel can be executed on. More...
|
|
Interface for calculating the final step of the Softmax Layer where each logit value is multiplied by the inverse of the sum of the logits.
Definition at line 79 of file CLSoftmaxLayerKernel.h.