fused_config_t< T > Struct Template Reference

#include <kernel_func.hpp>

Public Attributes
uint32_t	input_size
uint32_t	hidden_size
uint32_t	batch_size
uint32_t	sequence_length = 1
T *	layer_ptr = nullptr
T *	hx_ptr = nullptr
	layer_input = sequence_length x batch_size x input_size
T *	W_ir_ptr = nullptr
	h_x input = batch_size x hidden_size
T *	W_hr_ptr = nullptr
T *	W_iz_ptr = nullptr
T *	W_hz_ptr = nullptr
T *	W_in_ptr = nullptr
T *	W_hn_ptr = nullptr
T *	cell_out_ptr = nullptr
T *	layer_output = nullptr
	cell output = sequence_length x batch_size x hidden_size
T *	one_cell_ptr = nullptr
	layer_output = layer_size x batch_size x hidden_size

Member Data Documentation

batch_size

template<typename T >

uint32_t fused_config_t< T >::batch_size

cell_out_ptr

template<typename T >

T* fused_config_t< T >::cell_out_ptr = nullptr

hidden_size

template<typename T >

uint32_t fused_config_t< T >::hidden_size

hx_ptr

template<typename T >

T* fused_config_t< T >::hx_ptr = nullptr

layer_input = sequence_length x batch_size x input_size

input_size

template<typename T >

uint32_t fused_config_t< T >::input_size

layer_output

template<typename T >

T* fused_config_t< T >::layer_output = nullptr

cell output = sequence_length x batch_size x hidden_size

layer_ptr

template<typename T >

T* fused_config_t< T >::layer_ptr = nullptr

one_cell_ptr

template<typename T >

T* fused_config_t< T >::one_cell_ptr = nullptr

layer_output = layer_size x batch_size x hidden_size

sequence_length

template<typename T >

uint32_t fused_config_t< T >::sequence_length = 1

W_hn_ptr

template<typename T >

T* fused_config_t< T >::W_hn_ptr = nullptr

W_hr_ptr

template<typename T >

T* fused_config_t< T >::W_hr_ptr = nullptr

W_hz_ptr

template<typename T >

T* fused_config_t< T >::W_hz_ptr = nullptr

W_in_ptr

template<typename T >

T* fused_config_t< T >::W_in_ptr = nullptr

W_ir_ptr

template<typename T >

T* fused_config_t< T >::W_ir_ptr = nullptr

h_x input = batch_size x hidden_size

W_iz_ptr

template<typename T >

T* fused_config_t< T >::W_iz_ptr = nullptr