slice_tile.hpp File Reference

slice_tile.hpp File Reference#

Composable Kernel: slice_tile.hpp File Reference
slice_tile.hpp File Reference

Go to the source code of this file.

Namespaces

namespace  ck_tile

Functions

template<typename BottomTensorView_, typename WindowLengths_, index_t... SliceBegins, index_t... SliceEnds>
CK_TILE_DEVICE constexpr auto ck_tile::get_slice_tile (const tile_window_with_static_lengths< BottomTensorView_, WindowLengths_ > &tile, sequence< SliceBegins... > slice_begins, sequence< SliceEnds... > slice_ends)
template<typename DataType_, typename StaticTileDistribution_, index_t... SliceBegins, index_t... SliceEnds>
CK_TILE_DEVICE constexpr auto ck_tile::get_slice_tile (const static_distributed_tensor< DataType_, StaticTileDistribution_ > &tile, sequence< SliceBegins... > slice_begins, sequence< SliceEnds... > slice_ends)
template<typename DstDataType_, typename DstStaticTileDistribution_, typename SrcDataType_, typename SrcStaticTileDistribution_, index_t... SliceBegins, index_t... SliceEnds>
CK_TILE_DEVICE constexpr auto ck_tile::set_slice_tile (static_distributed_tensor< DstDataType_, DstStaticTileDistribution_ > &dst_tile, const static_distributed_tensor< SrcDataType_, SrcStaticTileDistribution_ > &src_tile, sequence< SliceBegins... > slice_begins, sequence< SliceEnds... > slice_ends)