XeTLA v0.3.6
IntelĀ® Xe Templates for Linear Algebra - API Definition Document
 
Loading...
Searching...
No Matches
gate_recurrent_unit.cpp File Reference
#include "kernel_func.hpp"
#include "tests/utils/utils.hpp"
Include dependency graph for gate_recurrent_unit.cpp:

Functions

template<typename data_type >
int validation (data_type *layer_inputs_device, data_type *h0_inputs_device, std::vector< data_type * > i_weights, std::vector< data_type * > h_weights, data_type *hidden_outputs_device, data_type *layer_outputs_device, uint32_t batch_size, uint32_t input_size, uint32_t hidden_size, uint32_t sequence_length, sycl::queue &queue, uint32_t layer_size=1)
 
template<typename gru_config >
void gru_run (uint32_t iter)
 
int main ()
 

Function Documentation

◆ gru_run()

template<typename gru_config >
void gru_run ( uint32_t  iter)

malloc for inputs

◆ main()

int main ( )

◆ validation()

template<typename data_type >
int validation ( data_type *  layer_inputs_device,
data_type *  h0_inputs_device,
std::vector< data_type * >  i_weights,
std::vector< data_type * >  h_weights,
data_type *  hidden_outputs_device,
data_type *  layer_outputs_device,
uint32_t  batch_size,
uint32_t  input_size,
uint32_t  hidden_size,
uint32_t  sequence_length,
sycl::queue &  queue,
uint32_t  layer_size = 1 
)