21.02
|
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... | |
Descriptor used by the softmax kernels.
Definition at line 112 of file KernelDescriptors.h.
float beta { 1.f } |
A scaling factor for the exponent with default value 1.0.
Definition at line 114 of file KernelDescriptors.h.
Referenced by CLLogits1DMaxShiftExpSumKernel::configure(), CLSoftmaxLayerGeneric< IS_LOG >::configure(), CLLogits1DNormKernel::configure(), and CLSoftmaxLayerGeneric< IS_LOG >::validate().
DataType input_data_type { DataType::UNKNOWN } |
Input tensor data type.
Definition at line 116 of file KernelDescriptors.h.
Referenced by CLLogits1DMaxShiftExpSumKernel::configure(), CLSoftmaxLayerGeneric< IS_LOG >::configure(), and CLLogits1DNormKernel::configure().
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().