#include <multi_index_transform.hpp>
|
__host__ constexpr __device__ | RightPad ()=default |
|
__host__ constexpr __device__ | RightPad (const LowLength &low_length, const RightPadLength &right_pad_length) |
|
__host__ constexpr __device__ const auto & | GetUpperLengths () const |
|
template<typename UpIdx > |
__host__ constexpr __device__ bool | IsValidUpperIndexMappedToValidLowerIndex (const UpIdx &idx_up) const |
|
__host__ __device__ void | Print () const |
|
|
__host__ static constexpr __device__ index_t | GetNumOfLowerDimension () |
|
__host__ static constexpr __device__ index_t | GetNumOfUpperDimension () |
|
template<typename LowIdx , typename UpIdx > |
__host__ static constexpr __device__ void | CalculateLowerIndex (LowIdx &idx_low, const UpIdx &idx_up) |
|
template<typename LowIdxDiff , typename UpIdxDiff , typename LowIdx , typename UpIdx , index_t Hack> |
__host__ static __device__ void | UpdateLowerIndex (LowIdxDiff &idx_diff_low, const UpIdxDiff &idx_diff_up, LowIdx &idx_low, const UpIdx &, Number< Hack >) |
|
__host__ static constexpr __device__ bool | IsLinearTransform () |
|
__host__ static constexpr __device__ bool | IsValidUpperIndexAlwaysMappedToValidLowerIndex () |
|
__host__ static constexpr __device__ bool | IsKnownAtCompileTime () |
|
◆ LowerIndex
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
◆ UpLengths
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
◆ UpperIndex
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
◆ RightPad() [1/2]
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
__host__ constexpr __device__ ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::RightPad |
( |
| ) |
|
|
constexprdefault |
◆ RightPad() [2/2]
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
__host__ constexpr __device__ ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::RightPad |
( |
const LowLength & |
low_length, |
|
|
const RightPadLength & |
right_pad_length |
|
) |
| |
|
inlineconstexpr |
◆ CalculateLowerIndex()
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
template<typename LowIdx , typename UpIdx >
__host__ static constexpr __device__ void ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::CalculateLowerIndex |
( |
LowIdx & |
idx_low, |
|
|
const UpIdx & |
idx_up |
|
) |
| |
|
inlinestaticconstexpr |
◆ GetNumOfLowerDimension()
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
__host__ static constexpr __device__ index_t ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::GetNumOfLowerDimension |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ GetNumOfUpperDimension()
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
__host__ static constexpr __device__ index_t ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::GetNumOfUpperDimension |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ GetUpperLengths()
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
__host__ constexpr __device__ const auto& ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::GetUpperLengths |
( |
| ) |
const |
|
inlineconstexpr |
◆ IsKnownAtCompileTime()
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
__host__ static constexpr __device__ bool ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::IsKnownAtCompileTime |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ IsLinearTransform()
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
__host__ static constexpr __device__ bool ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::IsLinearTransform |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ IsValidUpperIndexAlwaysMappedToValidLowerIndex()
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
__host__ static constexpr __device__ bool ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::IsValidUpperIndexAlwaysMappedToValidLowerIndex |
( |
| ) |
|
|
inlinestaticconstexpr |
◆ IsValidUpperIndexMappedToValidLowerIndex()
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
template<typename UpIdx >
__host__ constexpr __device__ bool ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::IsValidUpperIndexMappedToValidLowerIndex |
( |
const UpIdx & |
idx_up | ) |
const |
|
inlineconstexpr |
◆ Print()
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
__host__ __device__ void ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::Print |
( |
| ) |
const |
|
inline |
◆ UpdateLowerIndex()
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
template<typename LowIdxDiff , typename UpIdxDiff , typename LowIdx , typename UpIdx , index_t Hack>
__host__ static __device__ void ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::UpdateLowerIndex |
( |
LowIdxDiff & |
idx_diff_low, |
|
|
const UpIdxDiff & |
idx_diff_up, |
|
|
LowIdx & |
idx_low, |
|
|
const UpIdx & |
, |
|
|
Number< Hack > |
|
|
) |
| |
|
inlinestatic |
◆ low_length_
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
LowLength ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::low_length_ |
◆ right_pad_length_
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
RightPadLength ck::RightPad< LowLength, RightPadLength, SkipIsValidCheck >::right_pad_length_ |
◆ up_lengths_
template<typename LowLength , typename RightPadLength , bool SkipIsValidCheck = false>
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/multi_index_transform.hpp