PoolKernelArgs< TensorShape, WindowShape > Struct Template Reference

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

Composable Kernel: ck_tile::PoolKernelArgs< TensorShape, WindowShape > Struct Template Reference
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: