OperatorTraits.hpp File Reference

#include <lbann/operators/operator.hpp>

Include dependency graph for OperatorTraits.hpp:

Go to the source code of this file.

Classes
struct	lbann::OperatorTraits< OpT >
struct	lbann::OperatorTraits< Operator< InputT, OutputT, D > >

Namespaces
	lbann

Typedefs
template<typename T , El::Device D>
using	lbann::DataParallelMatrixType = El::DistMatrix< T, El::Dist::STAR, El::Dist::VC, El::DistWrap::ELEMENT, D >
	The data type for data-parallel computation. More...
template<typename T , El::Device D>
using	lbann::ModelParallelMatrixType = El::DistMatrix< T, El::Dist::MC, El::Dist::MR, El::DistWrap::ELEMENT, D >
template<typename OpT >
using	lbann::InputValueType = typename OperatorTraits< OpT >::input_value_type
template<typename OpT >
using	lbann::OutputValueType = typename OperatorTraits< OpT >::output_value_type
template<typename OpT >
using	lbann::BaseOperatorType = typename OperatorTraits< OpT >::base_type
template<typename OpT >
using	lbann::InputDataParallelMatType = typename OperatorTraits< OpT >::input_data_parallel_mat_type
template<typename OpT >
using	lbann::OutputDataParallelMatType = typename OperatorTraits< OpT >::output_data_parallel_mat_type
template<typename OpT >
using	lbann::InputModelParallelMatType = typename OperatorTraits< OpT >::input_model_parallel_mat_type
template<typename OpT >
using	lbann::OutputModelParallelMatType = typename OperatorTraits< OpT >::output_model_parallel_mat_type
template<typename OpT >
using	lbann::InputTensorType = typename OperatorTraits< OpT >::input_tensor_type
template<typename OpT >
using	lbann::OutputTensorType = typename OperatorTraits< OpT >::output_tensor_type
template<typename OpT >
using	lbann::InputConstTensorType = typename OperatorTraits< OpT >::input_const_tensor_type
template<typename OpT >
using	lbann::OutputConstTensorType = typename OperatorTraits< OpT >::output_const_tensor_type

Variables
template<typename OpT >
constexpr El::Device	lbann::Device = OperatorTraits<OpT>::device