#include <tensor_adaptor.hpp>
|
| __host__ __device__ constexpr const auto & | GetTransforms () const |
| __host__ __device__ constexpr | TensorAdaptor () |
| __host__ __device__ constexpr | TensorAdaptor (const Transforms &transforms) |
| __host__ __device__ constexpr auto | GetElementSize () const |
| template<typename TopIdx> |
| __host__ __device__ constexpr 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__ __device__ constexpr ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::TensorAdaptor |
( |
| ) |
|
|
inlineconstexpr |
◆ TensorAdaptor() [2/2]
template<typename Transforms, typename LowerDimensionHiddenIdss, typename UpperDimensionHiddenIdss, typename BottomDimensionHiddenIds, typename TopDimensionHiddenIds>
| __host__ __device__ constexpr 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__ __device__ constexpr 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 __device__ constexpr auto ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetBottomDimensionHiddenIds |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ GetElementSize()
template<typename Transforms, typename LowerDimensionHiddenIdss, typename UpperDimensionHiddenIdss, typename BottomDimensionHiddenIds, typename TopDimensionHiddenIds>
| __host__ __device__ constexpr 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 __device__ constexpr auto ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetLowerDimensionHiddenIdss |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ GetNumOfBottomDimension()
template<typename Transforms, typename LowerDimensionHiddenIdss, typename UpperDimensionHiddenIdss, typename BottomDimensionHiddenIds, typename TopDimensionHiddenIds>
| __host__ static __device__ constexpr 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 __device__ constexpr 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 __device__ constexpr 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 __device__ constexpr 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 __device__ constexpr auto ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetTopDimensionHiddenIds |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ GetTransformAndItsUpperDimension()
template<typename Transforms, typename LowerDimensionHiddenIdss, typename UpperDimensionHiddenIdss, typename BottomDimensionHiddenIds, typename TopDimensionHiddenIds>
| __host__ static __device__ constexpr 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__ __device__ constexpr 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 __device__ constexpr auto ck::TensorAdaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >::GetUpperDimensionHiddenIdss |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ InitializeElementSize()
template<typename Transforms, typename LowerDimensionHiddenIdss, typename UpperDimensionHiddenIdss, typename BottomDimensionHiddenIds, typename TopDimensionHiddenIds>
| __host__ static __device__ constexpr 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 __device__ constexpr 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: