TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds > Struct Template Reference

TensorAdaptor&lt; Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds &gt; Struct Template Reference#

Composable Kernel: ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds > Struct Template Reference
ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds > Struct Template Reference

#include <tensor_adaptor.hpp>

Public Types

using HiddenIndex = MultiIndex< ndim_hidden_ >
 
using BottomIndex = MultiIndex< ndim_bottom_ >
 
using TopIndex = MultiIndex< ndim_top_ >
 
using ElementSize = remove_cv_t< decltype(InitializeElementSize(Transforms{}))>
 

Public Member Functions

__host__ constexpr __device__ const auto & GetTransforms () const
 
__host__ constexpr __device__ TensorAdaptor ()
 
__host__ constexpr __device__ TensorAdaptor (const Transforms &transforms)
 
__host__ constexpr __device__ auto GetElementSize () const
 
template<typename TopIdx >
__host__ constexpr __device__ auto CalculateBottomIndex (const TopIdx &idx_top) const
 
__host__ __device__ void Print () const
 

Static Public Member Functions

__host__ static constexpr __device__ index_t GetNumOfTransform ()
 
__host__ static constexpr __device__ auto GetLowerDimensionHiddenIdss ()
 
__host__ static constexpr __device__ auto GetUpperDimensionHiddenIdss ()
 
__host__ static constexpr __device__ auto GetTopDimensionHiddenIds ()
 
__host__ static constexpr __device__ auto GetBottomDimensionHiddenIds ()
 
__host__ static constexpr __device__ auto InitializeElementSize (const Transforms &transforms)
 
template<index_t IDim>
__host__ static constexpr __device__ auto GetTransformAndItsUpperDimension (Number< IDim >)
 
__host__ static constexpr __device__ index_t GetNumOfBottomDimension ()
 
__host__ static constexpr __device__ index_t GetNumOfTopDimension ()
 
__host__ static constexpr __device__ index_t GetNumOfHiddenDimension ()
 
__host__ static constexpr __device__ bool IsKnownAtCompileTime ()
 

Static Public Attributes

constexpr static index_t ntransform_ = GetNumOfTransform()
 
constexpr static index_t ndim_hidden_ = GetNumOfHiddenDimension()
 
constexpr static index_t ndim_bottom_ = GetNumOfBottomDimension()
 
constexpr static index_t ndim_top_ = GetNumOfTopDimension()
 

Member Typedef Documentation

◆ BottomIndex

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
using ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::BottomIndex = MultiIndex<ndim_bottom_>

◆ ElementSize

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
using ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::ElementSize = remove_cv_t<decltype(InitializeElementSize(Transforms{}))>

◆ HiddenIndex

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
using ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::HiddenIndex = MultiIndex<ndim_hidden_>

◆ TopIndex

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
using ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::TopIndex = MultiIndex<ndim_top_>

Constructor & Destructor Documentation

◆ TensorAdaptor() [1/2]

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ constexpr __device__ ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::TensorAdaptor ( )
inlineconstexpr

◆ TensorAdaptor() [2/2]

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ constexpr __device__ ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::TensorAdaptor ( const Transforms &  transforms)
inlineconstexpr

Member Function Documentation

◆ CalculateBottomIndex()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
template<typename TopIdx >
__host__ constexpr __device__ auto ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::CalculateBottomIndex ( const TopIdx &  idx_top) const
inlineconstexpr

◆ GetBottomDimensionHiddenIds()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ static constexpr __device__ auto ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetBottomDimensionHiddenIds ( )
inlinestaticconstexpr

◆ GetElementSize()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ constexpr __device__ auto ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetElementSize ( ) const
inlineconstexpr

◆ GetLowerDimensionHiddenIdss()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ static constexpr __device__ auto ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetLowerDimensionHiddenIdss ( )
inlinestaticconstexpr

◆ GetNumOfBottomDimension()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ static constexpr __device__ index_t ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetNumOfBottomDimension ( )
inlinestaticconstexpr

◆ GetNumOfHiddenDimension()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ static constexpr __device__ index_t ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetNumOfHiddenDimension ( )
inlinestaticconstexpr

◆ GetNumOfTopDimension()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ static constexpr __device__ index_t ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetNumOfTopDimension ( )
inlinestaticconstexpr

◆ GetNumOfTransform()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ static constexpr __device__ index_t ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetNumOfTransform ( )
inlinestaticconstexpr

◆ GetTopDimensionHiddenIds()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ static constexpr __device__ auto ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetTopDimensionHiddenIds ( )
inlinestaticconstexpr

◆ GetTransformAndItsUpperDimension()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
template<index_t IDim>
__host__ static constexpr __device__ auto ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetTransformAndItsUpperDimension ( Number< IDim >  )
inlinestaticconstexpr

◆ GetTransforms()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ constexpr __device__ const auto& ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetTransforms ( ) const
inlineconstexpr

◆ GetUpperDimensionHiddenIdss()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ static constexpr __device__ auto ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetUpperDimensionHiddenIdss ( )
inlinestaticconstexpr

◆ InitializeElementSize()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ static constexpr __device__ auto ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::InitializeElementSize ( const Transforms &  transforms)
inlinestaticconstexpr

◆ IsKnownAtCompileTime()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ static constexpr __device__ bool ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::IsKnownAtCompileTime ( )
inlinestaticconstexpr

◆ Print()

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
__host__ __device__ void ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::Print ( ) const
inline

Member Data Documentation

◆ ndim_bottom_

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
constexpr static index_t ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::ndim_bottom_ = GetNumOfBottomDimension()
staticconstexpr

◆ ndim_hidden_

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
constexpr static index_t ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::ndim_hidden_ = GetNumOfHiddenDimension()
staticconstexpr

◆ ndim_top_

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
constexpr static index_t ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::ndim_top_ = GetNumOfTopDimension()
staticconstexpr

◆ ntransform_

template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
constexpr static index_t ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::ntransform_ = GetNumOfTransform()
staticconstexpr

The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_description/tensor_adaptor.hpp