Compute Library
 21.11
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...
 
int32_t axis { 0 }
 The dimension in which to apply softmax. More...
 

Detailed Description

Descriptor used by the softmax kernels.

Definition at line 113 of file KernelDescriptors.h.

Field Documentation

◆ axis

int32_t axis { 0 }

The dimension in which to apply softmax.

Definition at line 118 of file KernelDescriptors.h.

Referenced by ClSoftmax::configure(), arm_compute::operator<<(), and ClSoftmax::validate().

◆ beta

float beta { 1.f }

A scaling factor for the exponent with default value 1.0.

Definition at line 115 of file KernelDescriptors.h.

Referenced by ClLogits1DMaxShiftExpSumKernel::configure(), ClLogits1DNormKernel::configure(), arm_compute::operator<<(), and ClSoftmax::validate().

◆ input_data_type

DataType input_data_type { DataType::UNKNOWN }

◆ is_log

bool is_log { false }

Flag used to perform Log Softmax operation.

Definition at line 116 of file KernelDescriptors.h.

Referenced by ClLogits1DMaxShiftExpSumKernel::configure(), ClLogits1DNormKernel::configure(), and arm_compute::operator<<().


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