Compute Library
 21.02
SoftmaxKernelInfo Struct Reference

Descriptor used by the softmax kernels. More...

#include <KernelDescriptors.h>

Data Fields

float beta { 1.f }
 A scaling factor for the exponent with default value 1.0. More...
 
bool is_log { false }
 Flag used to perform Log Softmax operation. More...
 
DataType input_data_type { DataType::UNKNOWN }
 Input tensor data type. More...
 

Detailed Description

Descriptor used by the softmax kernels.

Definition at line 112 of file KernelDescriptors.h.

Field Documentation

◆ beta

float beta { 1.f }

◆ input_data_type

◆ is_log

bool is_log { false }

Flag used to perform Log Softmax operation.

Definition at line 115 of file KernelDescriptors.h.

Referenced by CLLogits1DMaxShiftExpSumKernel::configure(), CLSoftmaxLayerGeneric< IS_LOG >::configure(), and CLLogits1DNormKernel::configure().


The documentation for this struct was generated from the following file: