csinn_one_hot_params Struct Reference

#include <csinn_data_structure.h>

Public Attributes
struct csinn_params_base	base
float	f_on_value
float	f_off_value
int32_t	on_value
int32_t	off_value
int32_t	depth
int32_t	axis

Detailed Description

CSI-NN one_hot params

Member Data Documentation

base

struct csinn_params_base csinn_one_hot_params::base

The basic information of the operator

f_on_value

float csinn_one_hot_params::f_on_value

float on value

f_off_value

float csinn_one_hot_params::f_off_value

float off value

on_value

int32_t csinn_one_hot_params::on_value

int on value

off_value

int32_t csinn_one_hot_params::off_value

int off value

depth

int32_t csinn_one_hot_params::depth

The size of the output tensor

axis

int32_t csinn_one_hot_params::axis

Axis