dnn_enums.hpp File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
	lbann

Enumerations
enum	lbann::fwd_conv_alg {   lbann::fwd_conv_alg::IMPLICIT_GEMM, lbann::fwd_conv_alg::IMPLICIT_PRECOMP_GEMM, lbann::fwd_conv_alg::GEMM, lbann::fwd_conv_alg::DIRECT,   lbann::fwd_conv_alg::FFT, lbann::fwd_conv_alg::FFT_TILING, lbann::fwd_conv_alg::WINOGRAD, lbann::fwd_conv_alg::WINOGRAD_NONFUSED }
	Which forward convolution algorithm to use. More...
enum	lbann::bwd_data_conv_alg {   lbann::bwd_data_conv_alg::CUDNN_ALGO_0, lbann::bwd_data_conv_alg::CUDNN_ALGO_1, lbann::bwd_data_conv_alg::FFT, lbann::bwd_data_conv_alg::FFT_TILING,   lbann::bwd_data_conv_alg::WINOGRAD, lbann::bwd_data_conv_alg::WINOGRAD_NONFUSED, lbann::bwd_data_conv_alg::IMPLICIT_GEMM }
	Which backward convolution algorithm to use. More...
enum	lbann::bwd_filter_conv_alg {   lbann::bwd_filter_conv_alg::CUDNN_ALGO_0, lbann::bwd_filter_conv_alg::CUDNN_ALGO_1, lbann::bwd_filter_conv_alg::FFT, lbann::bwd_filter_conv_alg::CUDNN_ALGO_3,   lbann::bwd_filter_conv_alg::WINOGRAD, lbann::bwd_filter_conv_alg::WINOGRAD_NONFUSED, lbann::bwd_filter_conv_alg::FFT_TILING, lbann::bwd_filter_conv_alg::IMPLICIT_GEMM }
	Which backward convolution filter algorithm to use. More...
enum	lbann::lrn_mode { lbann::lrn_mode::CROSS_CHANNEL_DIM1 }
	Internal LBANN names for supported LRN layer modes. More...
enum	lbann::pooling_mode { lbann::pooling_mode::MAX, lbann::pooling_mode::AVERAGE_COUNT_INCLUDE_PADDING, lbann::pooling_mode::AVERAGE_COUNT_EXCLUDE_PADDING, lbann::pooling_mode::MAX_DETERMINISTIC }
	Which pooling mode to use. More...
enum	lbann::softmax_mode { lbann::softmax_mode::INVALID, lbann::softmax_mode::INSTANCE, lbann::softmax_mode::CHANNEL }
	Which tensor dimensions to apply softmax over. More...
enum	lbann::softmax_alg { lbann::softmax_alg::FAST, lbann::softmax_alg::ACCURATE, lbann::softmax_alg::LOG }
	Internal LBANN names for supported softmax algorithms. More...