ConvParam Struct Reference

• ck_tile
• conv
• ConvParam

Public Member Functions | Public Attributes | List of all members

ck_tile::conv::ConvParam Struct Reference

#include <convolution_parameter.hpp>

Inheritance diagram for ck_tile::conv::ConvParam:

Public Member Functions
	ConvParam (ck_tile::index_t n_dim, ck_tile::index_t group_count, ck_tile::index_t n_batch, ck_tile::index_t n_out_channels, ck_tile::index_t n_in_channels, const std::vector< ck_tile::index_t > &filters_len, const std::vector< ck_tile::index_t > &input_len, const std::vector< ck_tile::index_t > &strides, const std::vector< ck_tile::index_t > &dilations, const std::vector< ck_tile::index_t > &left_pads, const std::vector< ck_tile::index_t > &right_pads)
	ConvParam (ck_tile::long_index_t n_dim, ck_tile::long_index_t group_count, ck_tile::long_index_t n_batch, ck_tile::long_index_t n_out_channels, ck_tile::long_index_t n_in_channels, const std::vector< ck_tile::long_index_t > &filters_len, const std::vector< ck_tile::long_index_t > &input_len, const std::vector< ck_tile::long_index_t > &strides, const std::vector< ck_tile::long_index_t > &dilations, const std::vector< ck_tile::long_index_t > &left_pads, const std::vector< ck_tile::long_index_t > &right_pads)
std::vector< ck_tile::long_index_t >	GetOutputSpatialLengths () const
std::size_t	GetFlops () const
template<typename InDataType>
std::size_t	GetInputByte () const
template<typename WeiDataType>
std::size_t	GetWeightByte () const
template<typename OutDataType>
std::size_t	GetOutputByte () const
template<typename InDataType, typename WeiDataType, typename OutDataType>
std::size_t	GetByte () const

Public Attributes
ck_tile::long_index_t	num_dim_spatial_
ck_tile::long_index_t	G_
ck_tile::long_index_t	N_
ck_tile::long_index_t	K_
ck_tile::long_index_t	C_
std::vector< ck_tile::long_index_t >	filter_spatial_lengths_
std::vector< ck_tile::long_index_t >	input_spatial_lengths_
std::vector< ck_tile::long_index_t >	output_spatial_lengths_
std::vector< ck_tile::long_index_t >	conv_filter_strides_
std::vector< ck_tile::long_index_t >	conv_filter_dilations_
std::vector< ck_tile::long_index_t >	input_left_pads_
std::vector< ck_tile::long_index_t >	input_right_pads_

Constructor & Destructor Documentation

◆ ConvParam() [1/2]

ck_tile::conv::ConvParam::ConvParam ( ck_tile::index_t n_dim, ck_tile::index_t group_count, ck_tile::index_t n_batch, ck_tile::index_t n_out_channels, ck_tile::index_t n_in_channels, const std::vector< ck_tile::index_t > & filters_len, const std::vector< ck_tile::index_t > & input_len, const std::vector< ck_tile::index_t > & strides, const std::vector< ck_tile::index_t > & dilations, const std::vector< ck_tile::index_t > & left_pads, const std::vector< ck_tile::index_t > & right_pads )

inline

◆ ConvParam() [2/2]

ck_tile::conv::ConvParam::ConvParam ( ck_tile::long_index_t n_dim, ck_tile::long_index_t group_count, ck_tile::long_index_t n_batch, ck_tile::long_index_t n_out_channels, ck_tile::long_index_t n_in_channels, const std::vector< ck_tile::long_index_t > & filters_len, const std::vector< ck_tile::long_index_t > & input_len, const std::vector< ck_tile::long_index_t > & strides, const std::vector< ck_tile::long_index_t > & dilations, const std::vector< ck_tile::long_index_t > & left_pads, const std::vector< ck_tile::long_index_t > & right_pads )

inline

Member Function Documentation

◆ GetByte()

template<typename InDataType, typename WeiDataType, typename OutDataType>

std::size_t ck_tile::conv::ConvParam::GetByte ( ) const

inline

◆ GetFlops()

std::size_t ck_tile::conv::ConvParam::GetFlops ( ) const

inline

◆ GetInputByte()

template<typename InDataType>

std::size_t ck_tile::conv::ConvParam::GetInputByte ( ) const

inline

◆ GetOutputByte()

template<typename OutDataType>

std::size_t ck_tile::conv::ConvParam::GetOutputByte ( ) const

inline

◆ GetOutputSpatialLengths()

std::vector< ck_tile::long_index_t > ck_tile::conv::ConvParam::GetOutputSpatialLengths ( ) const

inline

◆ GetWeightByte()

template<typename WeiDataType>

std::size_t ck_tile::conv::ConvParam::GetWeightByte ( ) const

inline

Member Data Documentation

◆ C_

ck_tile::long_index_t ck_tile::conv::ConvParam::C_

◆ conv_filter_dilations_

std::vector<ck_tile::long_index_t> ck_tile::conv::ConvParam::conv_filter_dilations_

◆ conv_filter_strides_

std::vector<ck_tile::long_index_t> ck_tile::conv::ConvParam::conv_filter_strides_

◆ filter_spatial_lengths_

std::vector<ck_tile::long_index_t> ck_tile::conv::ConvParam::filter_spatial_lengths_

◆ G_

ck_tile::long_index_t ck_tile::conv::ConvParam::G_

◆ input_left_pads_

std::vector<ck_tile::long_index_t> ck_tile::conv::ConvParam::input_left_pads_

◆ input_right_pads_

std::vector<ck_tile::long_index_t> ck_tile::conv::ConvParam::input_right_pads_

◆ input_spatial_lengths_

std::vector<ck_tile::long_index_t> ck_tile::conv::ConvParam::input_spatial_lengths_

◆ K_

ck_tile::long_index_t ck_tile::conv::ConvParam::K_

◆ N_

ck_tile::long_index_t ck_tile::conv::ConvParam::N_

◆ num_dim_spatial_

ck_tile::long_index_t ck_tile::conv::ConvParam::num_dim_spatial_

◆ output_spatial_lengths_

std::vector<ck_tile::long_index_t> ck_tile::conv::ConvParam::output_spatial_lengths_

---

The documentation for this struct was generated from the following file:

• tile/host/convolution_parameter.hpp