SHL 2.2.x
|
#include <csinn_data_structure.h>
Public Attributes | |
struct csinn_params_base | base |
int32_t | axis |
float | minus_inf |
CSI-NN where softmax params
struct csinn_params_base csinn_where_softmax_params::base |
The basic information of the operator
int32_t csinn_where_softmax_params::axis |
The axis used to calculate softmax
float csinn_where_softmax_params::minus_inf |
Constant attribution minus inf