PoolHostArgs< TensorShape, WindowShape > Struct Template Reference

PoolHostArgs&lt; TensorShape, WindowShape &gt; Struct Template Reference#

Composable Kernel: ck_tile::PoolHostArgs< TensorShape, WindowShape > Struct Template Reference
Public Member Functions | Public Attributes | List of all members
ck_tile::PoolHostArgs< TensorShape, WindowShape > Struct Template Reference

Host arguments for pooling operations. More...

#include <pool_kernel.hpp>

Public Member Functions

CK_TILE_HOST PoolHostArgs (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_)
 

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::PoolHostArgs< TensorShape, WindowShape >

Host arguments for pooling operations.

Constructor & Destructor Documentation

◆ PoolHostArgs()

template<typename TensorShape , typename WindowShape >
CK_TILE_HOST ck_tile::PoolHostArgs< TensorShape, WindowShape >::PoolHostArgs ( 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_ 
)
inline

Member Data Documentation

◆ input_left_pads

template<typename TensorShape , typename WindowShape >
WindowShape ck_tile::PoolHostArgs< TensorShape, WindowShape >::input_left_pads

◆ input_ptr

template<typename TensorShape , typename WindowShape >
const void* ck_tile::PoolHostArgs< TensorShape, WindowShape >::input_ptr

◆ input_right_pads

template<typename TensorShape , typename WindowShape >
WindowShape ck_tile::PoolHostArgs< TensorShape, WindowShape >::input_right_pads

◆ input_shape

template<typename TensorShape , typename WindowShape >
TensorShape ck_tile::PoolHostArgs< TensorShape, WindowShape >::input_shape

◆ input_strides

template<typename TensorShape , typename WindowShape >
TensorShape ck_tile::PoolHostArgs< TensorShape, WindowShape >::input_strides

◆ output_index_ptr

template<typename TensorShape , typename WindowShape >
void* ck_tile::PoolHostArgs< TensorShape, WindowShape >::output_index_ptr

◆ output_ptr

template<typename TensorShape , typename WindowShape >
void* ck_tile::PoolHostArgs< TensorShape, WindowShape >::output_ptr

◆ output_shape

template<typename TensorShape , typename WindowShape >
TensorShape ck_tile::PoolHostArgs< TensorShape, WindowShape >::output_shape

◆ output_strides

template<typename TensorShape , typename WindowShape >
TensorShape ck_tile::PoolHostArgs< TensorShape, WindowShape >::output_strides

◆ window_dilations

template<typename TensorShape , typename WindowShape >
WindowShape ck_tile::PoolHostArgs< TensorShape, WindowShape >::window_dilations

◆ window_lengths

template<typename TensorShape , typename WindowShape >
WindowShape ck_tile::PoolHostArgs< TensorShape, WindowShape >::window_lengths

◆ window_strides

template<typename TensorShape , typename WindowShape >
WindowShape ck_tile::PoolHostArgs< 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