DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize > Struct Template Reference

DevicePool2dFwd_NHWC_NHWC&lt; InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize &gt; Struct Template Reference#

Composable Kernel: ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize > Struct Template Reference
Classes | Public Types | Public Member Functions | Static Public Member Functions | Static Public Attributes | List of all members
ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize > Struct Template Reference

#include <device_pool2d_fwd_nhwc_nhwc.hpp>

Inheritance diagram for ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >:
ck::tensor_operation::device::DevicePoolFwd< 4, 2, InDataType, OutDataType, IndexDataType, tensor_layout::convolution::NHWC, tensor_layout::convolution::NHWC, ReduceOpId, OutputIndex > ck::tensor_operation::device::BaseOperator

Classes

struct  Argument
struct  Invoker

Public Types

using ReduceOperation = typename reduce_binary_operator<ReduceOpId>::opType
using InElementwiseOperation
using AccElementwiseOperation
using ABGridDescs
using AGridDesc_M_K = remove_cvref_t<decltype(ABGridDescs{}[I0])>
using BGridDesc_M = remove_cvref_t<decltype(ABGridDescs{}[I1])>

Public Member Functions

bool IsSupportedArgument (const BaseArgument *p_arg) override
virtual std::unique_ptr< BaseArgument > MakeArgumentPointer (const void *p_in_dev, void *p_out_dev, void *p_out_indices_dev, std::vector< ck::index_t > input_nchw_lengths, std::vector< ck::index_t > window_yx_lengths, std::vector< ck::index_t > output_nchw_lengths, std::vector< ck::index_t > input_nchw_stride, std::vector< ck::index_t > output_nchw_stride, std::vector< ck::index_t > indices_nchw_stride, std::vector< ck::index_t > window_yx_strides, std::vector< ck::index_t > window_yx_dilations, std::vector< ck::index_t > input_left_hw_pads, std::vector< ck::index_t > input_right_hw_pads, std::vector< ck::index_t > pooling_dims) override
std::unique_ptr< BaseInvoker > MakeInvokerPointer () override
std::string GetTypeString () const override
Public Member Functions inherited from ck::tensor_operation::device::BaseOperator
 BaseOperator ()=default
 BaseOperator (const BaseOperator &)=default
BaseOperator & operator= (const BaseOperator &)=default
virtual std::string GetInstanceString () const
virtual std::string GetTypeIdName () const
virtual std::optional< std::string > GetObjectName () const
virtual std::optional< std::string > GetTemplateInfo () const
virtual std::string GetTypeIdHashCode () const
virtual size_t GetWorkSpaceSize (const BaseArgument *) const
virtual void SetWorkSpacePointer (BaseArgument *p_arg, void *p_workspace, const StreamConfig &=StreamConfig{}) const
virtual ~BaseOperator ()

Static Public Member Functions

static auto MakeABGridDescriptor_A_M_K_B_M (std::vector< ck::index_t > input_nchw_lengths, std::vector< ck::index_t > output_nchw_lengths, std::vector< ck::index_t > input_nchw_stride, std::vector< ck::index_t > output_nchw_stride, std::vector< ck::index_t > window_spatial_yx_lengths, std::vector< ck::index_t > window_yx_strides, std::vector< ck::index_t > window_yx_dilations, std::vector< ck::index_t > input_left_hw_pads, std::vector< ck::index_t > input_right_hw_pads)

Static Public Attributes

static constexpr auto I0 = Number<0>{}
static constexpr auto I1 = Number<1>{}
static constexpr index_t InOutRank = 4
static constexpr index_t WindowRank = 2
static constexpr ck::index_t M_BlockTileSize = MThreadClusterSize * MThreadSliceSize
static constexpr ck::index_t K_BlockTileSize = KThreadClusterSize * KThreadSliceSize

Member Typedef Documentation

◆ ABGridDescs

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
using ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::ABGridDescs
Initial value:
decltype(MakeABGridDescriptor_A_M_K_B_M({}, {}, {}, {}, {}, {}, {}, {}, {}))
ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC::MakeABGridDescriptor_A_M_K_B_M
static auto MakeABGridDescriptor_A_M_K_B_M(std::vector< ck::index_t > input_nchw_lengths, std::vector< ck::index_t > output_nchw_lengths, std::vector< ck::index_t > input_nchw_stride, std::vector< ck::index_t > output_nchw_stride, std::vector< ck::index_t > window_spatial_yx_lengths, std::vector< ck::index_t > window_yx_strides, std::vector< ck::index_t > window_yx_dilations, std::vector< ck::index_t > input_left_hw_pads, std::vector< ck::index_t > input_right_hw_pads)
Definition device_pool2d_fwd_nhwc_nhwc.hpp:62

◆ AccElementwiseOperation

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
using ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::AccElementwiseOperation
Initial value:
typename reduce_unary_operator<ReduceOpId, true, true>::AccElementwiseOperation
ck::reduce_unary_operator::AccElementwiseOperation
tensor_operation::element_wise::PassThrough AccElementwiseOperation
Definition reduction_operator_mapping.hpp:93

◆ AGridDesc_M_K

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
using ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::AGridDesc_M_K = remove_cvref_t<decltype(ABGridDescs{}[I0])>

◆ BGridDesc_M

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
using ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::BGridDesc_M = remove_cvref_t<decltype(ABGridDescs{}[I1])>

◆ InElementwiseOperation

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
using ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::InElementwiseOperation
Initial value:
typename reduce_unary_operator<ReduceOpId, true, true>::InElementwiseOperation
ck::reduce_unary_operator::InElementwiseOperation
tensor_operation::element_wise::PassThrough InElementwiseOperation
Definition reduction_operator_mapping.hpp:92

◆ ReduceOperation

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
using ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::ReduceOperation = typename reduce_binary_operator<ReduceOpId>::opType

Member Function Documentation

◆ GetTypeString()

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
std::string ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::GetTypeString ( ) const
inlineoverridevirtual

Reimplemented from ck::tensor_operation::device::BaseOperator.

◆ IsSupportedArgument()

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
bool ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::IsSupportedArgument ( const BaseArgument * p_arg)
inlineoverridevirtual

Reimplemented from ck::tensor_operation::device::BaseOperator.

◆ MakeABGridDescriptor_A_M_K_B_M()

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
auto ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::MakeABGridDescriptor_A_M_K_B_M ( std::vector< ck::index_t > input_nchw_lengths,
std::vector< ck::index_t > output_nchw_lengths,
std::vector< ck::index_t > input_nchw_stride,
std::vector< ck::index_t > output_nchw_stride,
std::vector< ck::index_t > window_spatial_yx_lengths,
std::vector< ck::index_t > window_yx_strides,
std::vector< ck::index_t > window_yx_dilations,
std::vector< ck::index_t > input_left_hw_pads,
std::vector< ck::index_t > input_right_hw_pads )
inlinestatic

◆ MakeArgumentPointer()

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
virtual std::unique_ptr< BaseArgument > ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::MakeArgumentPointer ( const void * p_in_dev,
void * p_out_dev,
void * p_out_indices_dev,
std::vector< ck::index_t > input_nchw_lengths,
std::vector< ck::index_t > window_yx_lengths,
std::vector< ck::index_t > output_nchw_lengths,
std::vector< ck::index_t > input_nchw_stride,
std::vector< ck::index_t > output_nchw_stride,
std::vector< ck::index_t > indices_nchw_stride,
std::vector< ck::index_t > window_yx_strides,
std::vector< ck::index_t > window_yx_dilations,
std::vector< ck::index_t > input_left_hw_pads,
std::vector< ck::index_t > input_right_hw_pads,
std::vector< ck::index_t > pooling_dims )
inlineoverridevirtual

Implements ck::tensor_operation::device::DevicePoolFwd< 4, 2, InDataType, OutDataType, IndexDataType, tensor_layout::convolution::NHWC, tensor_layout::convolution::NHWC, ReduceOpId, OutputIndex >.

◆ MakeInvokerPointer()

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
std::unique_ptr< BaseInvoker > ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::MakeInvokerPointer ( )
inlineoverridevirtual

Implements ck::tensor_operation::device::DevicePoolFwd< 4, 2, InDataType, OutDataType, IndexDataType, tensor_layout::convolution::NHWC, tensor_layout::convolution::NHWC, ReduceOpId, OutputIndex >.

Member Data Documentation

◆ I0

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
auto ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::I0 = Number<0>{}
staticconstexpr

◆ I1

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
auto ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::I1 = Number<1>{}
staticconstexpr

◆ InOutRank

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
index_t ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::InOutRank = 4
staticconstexpr

◆ K_BlockTileSize

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
ck::index_t ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::K_BlockTileSize = KThreadClusterSize * KThreadSliceSize
staticconstexpr

◆ M_BlockTileSize

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
ck::index_t ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::M_BlockTileSize = MThreadClusterSize * MThreadSliceSize
staticconstexpr

◆ WindowRank

template<typename InDataType, typename OutDataType, typename IndexDataType, typename ComputeDataType, ck::ReduceTensorOp ReduceOpId, bool OutputIndex, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
index_t ck::tensor_operation::device::DevicePool2dFwd_NHWC_NHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::WindowRank = 2
staticconstexpr
The documentation for this struct was generated from the following file: