Descriptor used by the softmax kernels. More...

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 135 of file KernelDescriptors.h.

Field Documentation

int32_t axis {0}

The dimension in which to apply softmax.

Definition at line 140 of file KernelDescriptors.h.

float beta {1.f}

A scaling factor for the exponent with default value 1.0.

Definition at line 137 of file KernelDescriptors.h.

Input tensor data type.

Definition at line 139 of file KernelDescriptors.h.

bool is_log {false}

Flag used to perform Log Softmax operation.

Definition at line 138 of file KernelDescriptors.h.

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