21.05
|
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... | |
int32_t | axis { 0 } |
The dimension in which to apply softmax. More... | |
Descriptor used by the softmax kernels.
Definition at line 112 of file KernelDescriptors.h.
int32_t axis { 0 } |
The dimension in which to apply softmax.
Definition at line 117 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.
DataType input_data_type { DataType::UNKNOWN } |
Input tensor data type.
Definition at line 116 of file KernelDescriptors.h.
bool is_log { false } |
Flag used to perform Log Softmax operation.
Definition at line 115 of file KernelDescriptors.h.