tensor_descriptor.hpp File Reference

• include
• ck
• tensor_description

Classes | Namespaces | Typedefs | Functions

tensor_descriptor.hpp File Reference

#include "ck/utility/common_header.hpp"
#include "ck/utility/sequence_helper.hpp"
#include "ck/tensor_description/multi_index_transform.hpp"

Go to the source code of this file.

Classes
struct	ck::TensorDescriptor< Transforms, LowerDimensionIdss, UpperDimensionIdss, VisibleDimensionIds, ElementSpaceSize >
struct	ck::TensorCoordinate< NDimHidden, VisibleDimensionIds >
struct	ck::TensorCoordinateStep< NTransform, NDimVisible, UpdateLowerIndexHack >
struct	ck::lambda_get_up_dim_num< NewTransforms >

Namespaces
namespace	ck

Typedefs
template<typename TensorDesc>
using	ck::TensorCoordinate_t
template<typename TensorDesc>
using	ck::TensorCoordinateStep_t

Functions
template<typename OldTensorDescriptor, typename NewTransforms, typename NewLowerDimensionOldVisibleIdss, typename NewUpperDimensionNewVisibleIdss>
__host__ __device__ constexpr auto	ck::transform_tensor_descriptor (const OldTensorDescriptor &old_tensor_desc, const NewTransforms &new_transforms, NewLowerDimensionOldVisibleIdss, NewUpperDimensionNewVisibleIdss)
template<typename TensorDesc, typename VisibleIndex>
__host__ __device__ constexpr auto	ck::make_tensor_coordinate (const TensorDesc &tensor_desc, const VisibleIndex &idx_visible)
template<typename TensorDesc, typename VisibleIndex, typename UpdateLowerIndexHack>
__host__ __device__ constexpr auto	ck::make_tensor_coordinate_step (const TensorDesc &, const VisibleIndex &idx_diff_visible, UpdateLowerIndexHack)
template<typename TensorDesc, typename VisibleIndex>
__host__ __device__ constexpr auto	ck::make_tensor_coordinate_step (const TensorDesc &, const VisibleIndex &idx_diff_visible)
template<typename TensorDesc, typename TensorCoord, typename TensorCoordStep>
__host__ __device__ constexpr void	ck::move_tensor_coordinate (const TensorDesc &tensor_desc, TensorCoord &coord, const TensorCoordStep &coord_step)
template<typename TensorDesc, typename TensorCoord>
__host__ __device__ constexpr bool	ck::coordinate_has_valid_offset_assuming_visible_index_is_valid (const TensorDesc &tensor_desc, const TensorCoord &coord)
template<typename TensorDesc, typename TensorCoord>
__host__ __device__ constexpr bool	ck::coordinate_has_valid_offset (const TensorDesc &tensor_desc, const TensorCoord &coord)