ConvParam Struct Reference

• ck
• utils
• conv
• ConvParam

Public Member Functions | Public Attributes | List of all members

ck::utils::conv::ConvParam Struct Reference

#include <convolution_parameter.hpp>

Public Member Functions
	ConvParam ()
	ConvParam (ck::index_t n_dim, ck::index_t group_count, ck::index_t n_batch, ck::index_t n_out_channels, ck::index_t n_in_channels, const std::vector< ck::index_t > &filters_len, const std::vector< ck::index_t > &input_len, const std::vector< ck::index_t > &strides, const std::vector< ck::index_t > &dilations, const std::vector< ck::index_t > &left_pads, const std::vector< ck::index_t > &right_pads)
	ConvParam (ck::long_index_t n_dim, ck::long_index_t group_count, ck::long_index_t n_batch, ck::long_index_t n_out_channels, ck::long_index_t n_in_channels, const std::vector< ck::long_index_t > &filters_len, const std::vector< ck::long_index_t > &input_len, const std::vector< ck::long_index_t > &strides, const std::vector< ck::long_index_t > &dilations, const std::vector< ck::long_index_t > &left_pads, const std::vector< ck::long_index_t > &right_pads)
std::vector< ck::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::long_index_t	num_dim_spatial_
ck::long_index_t	G_
ck::long_index_t	N_
ck::long_index_t	K_
ck::long_index_t	C_
std::vector< ck::long_index_t >	filter_spatial_lengths_
std::vector< ck::long_index_t >	input_spatial_lengths_
std::vector< ck::long_index_t >	output_spatial_lengths_
std::vector< ck::long_index_t >	conv_filter_strides_
std::vector< ck::long_index_t >	conv_filter_dilations_
std::vector< ck::long_index_t >	input_left_pads_
std::vector< ck::long_index_t >	input_right_pads_

Constructor & Destructor Documentation

◆ ConvParam() [1/3]

ck::utils::conv::ConvParam::ConvParam

(

)

◆ ConvParam() [2/3]

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

◆ ConvParam() [3/3]

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

Member Function Documentation

◆ GetByte()

template<typename InDataType, typename WeiDataType, typename OutDataType>

std::size_t ck::utils::conv::ConvParam::GetByte ( ) const

inline

◆ GetFlops()

std::size_t ck::utils::conv::ConvParam::GetFlops

(

)

const

◆ GetInputByte()

template<typename InDataType>

std::size_t ck::utils::conv::ConvParam::GetInputByte ( ) const

inline

◆ GetOutputByte()

template<typename OutDataType>

std::size_t ck::utils::conv::ConvParam::GetOutputByte ( ) const

inline

◆ GetOutputSpatialLengths()

std::vector< ck::long_index_t > ck::utils::conv::ConvParam::GetOutputSpatialLengths

(

)

const

◆ GetWeightByte()

template<typename WeiDataType>

std::size_t ck::utils::conv::ConvParam::GetWeightByte ( ) const

inline

Member Data Documentation

◆ C_

ck::long_index_t ck::utils::conv::ConvParam::C_

◆ conv_filter_dilations_

std::vector<ck::long_index_t> ck::utils::conv::ConvParam::conv_filter_dilations_

◆ conv_filter_strides_

std::vector<ck::long_index_t> ck::utils::conv::ConvParam::conv_filter_strides_

◆ filter_spatial_lengths_

std::vector<ck::long_index_t> ck::utils::conv::ConvParam::filter_spatial_lengths_

◆ G_

ck::long_index_t ck::utils::conv::ConvParam::G_

◆ input_left_pads_

std::vector<ck::long_index_t> ck::utils::conv::ConvParam::input_left_pads_

◆ input_right_pads_

std::vector<ck::long_index_t> ck::utils::conv::ConvParam::input_right_pads_

◆ input_spatial_lengths_

std::vector<ck::long_index_t> ck::utils::conv::ConvParam::input_spatial_lengths_

◆ K_

ck::long_index_t ck::utils::conv::ConvParam::K_

◆ N_

ck::long_index_t ck::utils::conv::ConvParam::N_

◆ num_dim_spatial_

ck::long_index_t ck::utils::conv::ConvParam::num_dim_spatial_

◆ output_spatial_lengths_

std::vector<ck::long_index_t> ck::utils::conv::ConvParam::output_spatial_lengths_

---

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

• library/utility/convolution_parameter.hpp