tensor_adaptor.hpp File Reference

tensor_adaptor.hpp File Reference#

Composable Kernel: tensor_adaptor.hpp File Reference
tensor_adaptor.hpp File Reference

Go to the source code of this file.

Classes

struct  ck_tile::tensor_adaptor< Transforms, LowerDimensionHiddenIdss, UpperDimensionHiddenIdss, BottomDimensionHiddenIds, TopDimensionHiddenIds >
struct  ck_tile::lambda_get_up_dim_num< NewTransforms >

Namespaces

namespace  ck_tile

Macros

#define CONSTRUCT_TENSOR_ADAPTOR_FROM_ENCODING(encoded_tensor_adaptor)
#define CONSTRUCT_STATIC_TENSOR_ADAPTOR_FROM_ENCODING(encoded_tensor_adaptor)

Functions

template<typename Transforms, typename LowerDimensionOldTopIdss, typename UpperDimensionNewTopIdss>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::make_single_stage_tensor_adaptor (const Transforms &transforms, LowerDimensionOldTopIdss, UpperDimensionNewTopIdss)
template<typename OldTensorAdaptor, typename NewTransforms, typename NewLowerDimensionOldTopIdss, typename NewUpperDimensionNewTopIdss>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::transform_tensor_adaptor (const OldTensorAdaptor &old_tensor_adaptor, const NewTransforms &new_transforms, NewLowerDimensionOldTopIdss, NewUpperDimensionNewTopIdss)
template<typename TensorAdaptor0, typename TensorAdaptor1>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::chain_tensor_adaptors (const TensorAdaptor0 &adaptor0, const TensorAdaptor1 &adaptor1)
template<typename X, typename... Xs, typename std::enable_if< sizeof...(Xs) > = 2, bool, ::type = false>
CK_TILE_HOST_DEVICE constexpr auto ck_tile::chain_tensor_adaptors (const X &x, const Xs &... xs)

Macro Definition Documentation

◆ CONSTRUCT_STATIC_TENSOR_ADAPTOR_FROM_ENCODING

#define CONSTRUCT_STATIC_TENSOR_ADAPTOR_FROM_ENCODING ( encoded_tensor_adaptor)

◆ CONSTRUCT_TENSOR_ADAPTOR_FROM_ENCODING

#define CONSTRUCT_TENSOR_ADAPTOR_FROM_ENCODING ( encoded_tensor_adaptor)