tensor_descriptor.hpp File Reference

tensor_descriptor.hpp File Reference#

Composable Kernel: tensor_descriptor.hpp File Reference
Classes | Namespaces | Functions
tensor_descriptor.hpp File Reference
#include "ck_tile/core/config.hpp"
#include "ck_tile/core/numeric/integer.hpp"
#include "ck_tile/core/numeric/integral_constant.hpp"
#include "ck_tile/core/algorithm/coordinate_transform.hpp"
#include "ck_tile/core/tensor/tensor_adaptor.hpp"
#include "ck_tile/core/container/container_helper.hpp"
#include "ck_tile/core/container/multi_index.hpp"
#include "ck_tile/core/numeric/math.hpp"
#include "ck_tile/core/utility/type_traits.hpp"

Go to the source code of this file.

Classes

struct  ck_tile::tensor_descriptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, TopDimensionHiddenIds, ElementSpaceSize, GuaranteedVectorLengths_, GuaranteedVectorSrides_ >

Namespaces

namespace  ck_tile
namespace  ck_tile::detail

Functions

template<typename Adaptor, typename ElementSpaceSize>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::make_tensor_descriptor_from_adaptor (const Adaptor &adaptor, const ElementSpaceSize &element_space_size)
template<typename OldTensorDescriptor, typename NewTransforms, typename NewLowerDimensionOldTopIdss, typename NewUpperDimensionNewTopIdss>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::transform_tensor_descriptor (const OldTensorDescriptor &old_tensor_desc, const NewTransforms &new_transforms, NewLowerDimensionOldTopIdss, NewUpperDimensionNewTopIdss)
template<typename Lengths, typename Strides, index_t I, typename AccOld>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::detail::calculate_element_space_size_impl (const Lengths &lengths, const Strides &strides, number< I > i, AccOld acc_old)
template<typename... Lengths, typename... Strides, index_t GuaranteedLastDimensionVectorLength = -1, index_t GuaranteedLastDimensionVectorStride = -1, typename std::enable_if< sizeof...(Lengths)==sizeof...(Strides), bool >::type = false>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::make_naive_tensor_descriptor (const tuple< Lengths... > &lengths, const tuple< Strides... > &strides, number< GuaranteedLastDimensionVectorLength >=number<-1 >{}, number< GuaranteedLastDimensionVectorStride >=number<-1 >{})
template<typename... Lengths, typename... Strides, typename offset, index_t GuaranteedLastDimensionVectorLength = -1, index_t GuaranteedLastDimensionVectorStride = -1, typename std::enable_if< sizeof...(Lengths)==sizeof...(Strides), bool >::type = false>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::make_naive_tensor_descriptor_with_offset (const tuple< Lengths... > &lengths, const tuple< Strides... > &strides, const offset &os, number< GuaranteedLastDimensionVectorLength >=number<-1 >{}, number< GuaranteedLastDimensionVectorStride >=number<-1 >{})
template<typename... Lengths, index_t GuaranteedLastDimensionVectorLength = -1>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::make_naive_tensor_descriptor_packed (const tuple< Lengths... > &lengths, number< GuaranteedLastDimensionVectorLength >=number<-1 >{})
template<typename... Lengths, typename... Strides, typename Offset, index_t GuaranteedLastDimensionVectorLength = -1, typename std::enable_if< sizeof...(Lengths)==sizeof...(Strides), bool >::type = false>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::make_naive_tensor_descriptor_packed_with_offset (const tuple< Lengths... > &lengths, const Offset &offset, number< GuaranteedLastDimensionVectorLength >=number<-1 >{})
template<typename... Lengths, typename Align>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::make_naive_tensor_descriptor_aligned (const tuple< Lengths... > &lengths, Align align)