kfac_block_bn.hpp File Reference

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

Include dependency graph for kfac_block_bn.hpp:

Go to the source code of this file.

Classes
class	lbann::kfac_block_bn< Device >

Namespaces
	lbann
	lbann::kfac_bn_util

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