kfac_block_fc_conv.hpp File Reference

#include "lbann/execution_algorithms/kfac/kfac_block.hpp"
#include "lbann/layers/learning/convolution.hpp"

Include dependency graph for kfac_block_fc_conv.hpp:

Go to the source code of this file.

Classes
class	lbann::kfac_block_fc_conv< Device >

Namespaces
	lbann
	lbann::kfac_fc_conv_util

Functions
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. More...
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. More...
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. More...