PoolHostArgs< TensorShape, WindowShape > Struct Template Reference

PoolHostArgs< TensorShape, WindowShape > Struct Template Reference#

Composable Kernel: 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:

pool_kernel.hpp