lbann::kfac_bn_util Namespace Reference

Functions
template<El::Device Device>
void	compute_bn_factor_data2col (const El::Matrix< DataType, Device > &activations, const El::Matrix< DataType, Device > &errors, const El::Matrix< DataType, Device > &scales, const El::Matrix< DataType, Device > &biases, El::Matrix< DataType, Device > &cols, size_t batch_size, size_t num_channels, size_t spatial_prod, const El::SyncInfo< Device > &sync_info)
	The memory copy part of compute_bn_factor. Combined with GEMM. More...

Function Documentation

compute_bn_factor_data2col()

template<El::Device Device>

void lbann::kfac_bn_util::compute_bn_factor_data2col	(	const El::Matrix< DataType, Device > &	activations,
		const El::Matrix< DataType, Device > &	errors,
		const El::Matrix< DataType, Device > &	scales,
		const El::Matrix< DataType, Device > &	biases,
		El::Matrix< DataType, Device > &	cols,
		size_t	batch_size,
		size_t	num_channels,
		size_t	spatial_prod,
		const El::SyncInfo< Device > &	sync_info
	)

The memory copy part of compute_bn_factor. Combined with GEMM.