LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
lbann::utils::TensorView< T, D > Class Template Reference

#include <tensor.hpp>

Inheritance diagram for lbann::utils::TensorView< T, D >:
[legend]
Collaboration diagram for lbann::utils::TensorView< T, D >:
[legend]

Public Member Functions

template<typename MatT >
 TensorView (MatT &&mat)
 
template<typename MatT >
 TensorView (MatT &&mat, std::vector< size_t > const &dims)
 
- Public Member Functions inherited from lbann::utils::details::MatrixAsTensorView< El::Matrix< T, D > >
 MatrixAsTensorView (MatT &&mat, std::vector< size_t > const &dims)
 
- Public Member Functions inherited from lbann::utils::details::MatrixReferenceWrapper< El::Matrix< T, D > >
 MatrixReferenceWrapper (MatT &&x)
 
 operator matrix_type & () const noexcept
 
matrix_typedata () const noexcept
 

Private Types

using base_type = details::MatrixAsTensorView< El::Matrix< T, D > >
 

Additional Inherited Members

- Public Types inherited from lbann::utils::details::MatrixReferenceWrapper< El::Matrix< T, D > >
using matrix_type = El::Matrix< T, D >
 
- Public Attributes inherited from lbann::utils::details::MatrixAsTensorView< El::Matrix< T, D > >
 m_dims
 

Detailed Description

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

Definition at line 188 of file tensor.hpp.

Member Typedef Documentation

◆ base_type

template<typename T, El::Device D>
using lbann::utils::TensorView< T, D >::base_type = details::MatrixAsTensorView<El::Matrix<T, D> >
private

Definition at line 190 of file tensor.hpp.

Constructor & Destructor Documentation

◆ TensorView() [1/2]

template<typename T, El::Device D>
template<typename MatT >
lbann::utils::TensorView< T, D >::TensorView ( MatT &&  mat)
inline

Definition at line 194 of file tensor.hpp.

Here is the call graph for this function:

◆ TensorView() [2/2]

template<typename T, El::Device D>
template<typename MatT >
lbann::utils::TensorView< T, D >::TensorView ( MatT &&  mat,
std::vector< size_t > const &  dims 
)
inline

Definition at line 198 of file tensor.hpp.


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