lbann::kfac_fc_conv_util Namespace Reference

Functions
template<El::Device Device>
void	get_diagonal (El::Matrix< DataType, Device > &diag, const El::Matrix< DataType, Device > &A, const El::SyncInfo< Device > &sync_info)
	Get diagonal elements of a matrix. More...
template<El::Device Device>
void	conv_transpose (const El::Matrix< DataType, Device > &activations, El::Matrix< DataType, Device > &act_columns, size_t mini_batch_size, size_t num_channels, size_t spatial_prod, const El::SyncInfo< Device > &sync_info)
	Transpose NC(D)HW matrix to N(D)HWC. More...
template<El::Device Device>
void	im2col (const El::Matrix< DataType, Device > &im, El::Matrix< DataType, Device > &col, const int num_channels, const int im_num_dims, const int *im_dims, const int *im_pads, const int *window_dims, const int *window_strides, const int batch_size, const El::SyncInfo< Device > &sync_info)
	im2col. More...

Function Documentation

conv_transpose()

template<El::Device Device>

void lbann::kfac_fc_conv_util::conv_transpose	(	const El::Matrix< DataType, Device > &	activations,
		El::Matrix< DataType, Device > &	act_columns,
		size_t	mini_batch_size,
		size_t	num_channels,
		size_t	spatial_prod,
		const El::SyncInfo< Device > &	sync_info
	)

Transpose NC(D)HW matrix to N(D)HWC.

get_diagonal()

template<El::Device Device>

void lbann::kfac_fc_conv_util::get_diagonal	(	El::Matrix< DataType, Device > &	diag,
		const El::Matrix< DataType, Device > &	A,
		const El::SyncInfo< Device > &	sync_info
	)

Get diagonal elements of a matrix.

im2col()

template<El::Device Device>

void lbann::kfac_fc_conv_util::im2col	(	const El::Matrix< DataType, Device > &	im,
		El::Matrix< DataType, Device > &	col,
		const int	num_channels,
		const int	im_num_dims,
		const int *	im_dims,
		const int *	im_pads,
		const int *	window_dims,
		const int *	window_strides,
		const int	batch_size,
		const El::SyncInfo< Device > &	sync_info
	)

im2col.