lbann::utils::DistTensorView< T, D > Class Template Reference

#include <tensor.hpp>

Inheritance diagram for lbann::utils::DistTensorView< T, D >:

Collaboration diagram for lbann::utils::DistTensorView< T, D >:

Public Member Functions
template<typename MatT >
	DistTensorView (MatT &&mat)
template<typename MatT >
	DistTensorView (MatT &&mat, std::vector< size_t > const &dims)
TensorView< T, D > const &	local_data () const noexcept
	Access the local tensor data. More...
Public Member Functions inherited from lbann::utils::details::MatrixAsTensorView< El::AbstractDistMatrix< T > >
	MatrixAsTensorView (MatT &&mat, std::vector< size_t > const &dims)
Public Member Functions inherited from lbann::utils::details::MatrixReferenceWrapper< El::AbstractDistMatrix< T > >
	MatrixReferenceWrapper (MatT &&x)
	operator matrix_type & () const noexcept
matrix_type &	data () const noexcept

Private Types
using	base_type = details::MatrixAsTensorView< El::AbstractDistMatrix< T > >

Private Attributes
TensorView< T, D >	m_local_data

Additional Inherited Members
Public Types inherited from lbann::utils::details::MatrixReferenceWrapper< El::AbstractDistMatrix< T > >
using	matrix_type = El::AbstractDistMatrix< T >
Public Attributes inherited from lbann::utils::details::MatrixAsTensorView< El::AbstractDistMatrix< T > >
	m_dims

Detailed Description

template<typename T, El::Device D>
class lbann::utils::DistTensorView< T, D >

Definition at line 233 of file tensor.hpp.

Member Typedef Documentation

base_type

template<typename T , El::Device D>

using lbann::utils::DistTensorView< T, D >::base_type = details::MatrixAsTensorView<El::AbstractDistMatrix<T> >

private

Definition at line 236 of file tensor.hpp.

Constructor & Destructor Documentation

DistTensorView() [1/2]

template<typename T , El::Device D>

template<typename MatT >

lbann::utils::DistTensorView< T, D >::DistTensorView ( MatT &&  mat )

inline

Definition at line 240 of file tensor.hpp.

Here is the call graph for this function:

DistTensorView() [2/2]

template<typename T , El::Device D>

template<typename MatT >

lbann::utils::DistTensorView< T, D >::DistTensorView ( MatT &&  mat, std::vector< size_t > const &  dims  )

inline

Definition at line 244 of file tensor.hpp.

Here is the call graph for this function:

Member Function Documentation

local_data()

template<typename T , El::Device D>

TensorView<T, D> const& lbann::utils::DistTensorView< T, D >::local_data ( ) const

inlinenoexcept

Access the local tensor data.

Definition at line 250 of file tensor.hpp.

Member Data Documentation

m_local_data

template<typename T , El::Device D>

TensorView<T, D> lbann::utils::DistTensorView< T, D >::m_local_data

private

Definition at line 253 of file tensor.hpp.

---

The documentation for this class was generated from the following file:

• tensor.hpp