#include <tensor_adaptor.hpp>
|
__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 |
|
◆ BottomIndex
template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
◆ ElementSize
template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
◆ HiddenIndex
template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
◆ TopIndex
template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
◆ 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 |
◆ 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 |
◆ ndim_bottom_
template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
◆ ndim_hidden_
template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
◆ ndim_top_
template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
◆ ntransform_
template<typename Transforms , typename LowerDimensionHiddenIdss , typename UpperDimensionHiddenIdss , typename BottomDimensionHiddenIds , typename TopDimensionHiddenIds >
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