Utils::detail::TensorContractionPlan Class Reference

#include <TensorContraction.h>

Collaboration diagram for Utils::detail::TensorContractionPlan:

Public Member Functions
template<class T>
void	Execute (const T *a, const T *b, T *out, bool allow, T *packing=nullptr) const
size_t	ExtraBytes () const
const std::vector< size_t > &	GetDims () const
size_t	GetPackingSize () const
size_t	GetSize () const
template<class Pairs>
bool	Prepare (const std::vector< size_t > &da, const std::vector< size_t > &db, const Pairs &pairs, size_t byteLimit)

Detailed Description

Definition at line 555 of file TensorContraction.h.

Member Function Documentation

Execute()

template<class T>

void Utils::detail::TensorContractionPlan::Execute ( const T * a, const T * b, T * out, bool allow, T * packing = nullptr ) const

inline

Definition at line 600 of file TensorContraction.h.

References Utils::detail::TensorExecuteContraction().

ExtraBytes()

size_t Utils::detail::TensorContractionPlan::ExtraBytes ( ) const

inline

Definition at line 591 of file TensorContraction.h.

GetDims()

const std::vector< size_t > & Utils::detail::TensorContractionPlan::GetDims ( ) const

inline

Definition at line 567 of file TensorContraction.h.

GetPackingSize()

size_t Utils::detail::TensorContractionPlan::GetPackingSize ( ) const

inline

Definition at line 578 of file TensorContraction.h.

References Utils::detail::TensorAddBytes(), and Utils::detail::TensorMatrixContiguous().

GetSize()

size_t Utils::detail::TensorContractionPlan::GetSize ( ) const

inline

Definition at line 572 of file TensorContraction.h.

Prepare()

template<class Pairs>

bool Utils::detail::TensorContractionPlan::Prepare ( const std::vector< size_t > & da, const std::vector< size_t > & db, const Pairs & pairs, size_t byteLimit )

inline

Definition at line 558 of file TensorContraction.h.

---

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

• TensorContraction.h