xetla_softmax_fwd_t< dtype_in_, dtype_out_, tile_shape_, mem_space_in_, mem_space_out_, SIMD_, thread_num_, softmax_size_ >::arguments_t Struct Reference

#include <softmax.hpp>

Public Attributes
uint32_t	data_in_base
uint32_t	data_out_base
dtype_in *	data_in_ptr
dtype_out *	data_out_ptr

Member Data Documentation

data_in_base

template<typename dtype_in_ , typename dtype_out_ , typename tile_shape_ , mem_space mem_space_in_, mem_space mem_space_out_, uint32_t SIMD_, uint32_t thread_num_, uint32_t softmax_size_>

uint32_t xetla_softmax_fwd_t< dtype_in_, dtype_out_, tile_shape_, mem_space_in_, mem_space_out_, SIMD_, thread_num_, softmax_size_ >::arguments_t::data_in_base

data_in_ptr

template<typename dtype_in_ , typename dtype_out_ , typename tile_shape_ , mem_space mem_space_in_, mem_space mem_space_out_, uint32_t SIMD_, uint32_t thread_num_, uint32_t softmax_size_>

dtype_in* xetla_softmax_fwd_t< dtype_in_, dtype_out_, tile_shape_, mem_space_in_, mem_space_out_, SIMD_, thread_num_, softmax_size_ >::arguments_t::data_in_ptr

data_out_base

template<typename dtype_in_ , typename dtype_out_ , typename tile_shape_ , mem_space mem_space_in_, mem_space mem_space_out_, uint32_t SIMD_, uint32_t thread_num_, uint32_t softmax_size_>

uint32_t xetla_softmax_fwd_t< dtype_in_, dtype_out_, tile_shape_, mem_space_in_, mem_space_out_, SIMD_, thread_num_, softmax_size_ >::arguments_t::data_out_base

data_out_ptr

template<typename dtype_in_ , typename dtype_out_ , typename tile_shape_ , mem_space mem_space_in_, mem_space mem_space_out_, uint32_t SIMD_, uint32_t thread_num_, uint32_t softmax_size_>

dtype_out* xetla_softmax_fwd_t< dtype_in_, dtype_out_, tile_shape_, mem_space_in_, mem_space_out_, SIMD_, thread_num_, softmax_size_ >::arguments_t::data_out_ptr