PoolKernelArgs< TensorShape, WindowShape > Struct Template Reference

• ck_tile
• PoolKernelArgs

Public Attributes | List of all members

ck_tile::PoolKernelArgs< TensorShape, WindowShape > Struct Template Reference

Kernel arguments for pooling operations. More...

#include <pool_kernel.hpp>

Public Attributes
const void *	input_ptr
void *	output_ptr
void *	output_index_ptr
TensorShape	input_shape
TensorShape	output_shape
TensorShape	input_strides
TensorShape	output_strides
WindowShape	window_lengths
WindowShape	window_strides
WindowShape	window_dilations
WindowShape	input_left_pads
WindowShape	input_right_pads

Detailed Description

template<typename TensorShape, typename WindowShape>
struct ck_tile::PoolKernelArgs< TensorShape, WindowShape >

Kernel arguments for pooling operations.

Member Data Documentation

◆ input_left_pads

template<typename TensorShape , typename WindowShape >

WindowShape ck_tile::PoolKernelArgs< TensorShape, WindowShape >::input_left_pads

◆ input_ptr

template<typename TensorShape , typename WindowShape >

const void* ck_tile::PoolKernelArgs< TensorShape, WindowShape >::input_ptr

◆ input_right_pads

template<typename TensorShape , typename WindowShape >

WindowShape ck_tile::PoolKernelArgs< TensorShape, WindowShape >::input_right_pads

◆ input_shape

template<typename TensorShape , typename WindowShape >

TensorShape ck_tile::PoolKernelArgs< TensorShape, WindowShape >::input_shape

◆ input_strides

template<typename TensorShape , typename WindowShape >

TensorShape ck_tile::PoolKernelArgs< TensorShape, WindowShape >::input_strides

◆ output_index_ptr

template<typename TensorShape , typename WindowShape >

void* ck_tile::PoolKernelArgs< TensorShape, WindowShape >::output_index_ptr

◆ output_ptr

template<typename TensorShape , typename WindowShape >

void* ck_tile::PoolKernelArgs< TensorShape, WindowShape >::output_ptr

◆ output_shape

template<typename TensorShape , typename WindowShape >

TensorShape ck_tile::PoolKernelArgs< TensorShape, WindowShape >::output_shape

◆ output_strides

template<typename TensorShape , typename WindowShape >

TensorShape ck_tile::PoolKernelArgs< TensorShape, WindowShape >::output_strides

◆ window_dilations

template<typename TensorShape , typename WindowShape >

WindowShape ck_tile::PoolKernelArgs< TensorShape, WindowShape >::window_dilations

◆ window_lengths

template<typename TensorShape , typename WindowShape >

WindowShape ck_tile::PoolKernelArgs< TensorShape, WindowShape >::window_lengths

◆ window_strides

template<typename TensorShape , typename WindowShape >

WindowShape ck_tile::PoolKernelArgs< TensorShape, WindowShape >::window_strides

---

The documentation for this struct was generated from the following file:

• /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/pooling/kernel/pool_kernel.hpp