Argument Struct Reference

Argument Struct Reference#

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

#include <device_pool3d_fwd_ndhwc_ndhwc.hpp>

Inheritance diagram for ck::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument:
ck::tensor_operation::device::BaseArgument

Public Member Functions

 Argument (const InDataType *p_in_dev, OutDataType *p_out_dev, IndexDataType *p_out_indices_dev, std::vector< ck::index_t > &input_ncdhw_lengths, std::vector< ck::index_t > &output_ncdhw_lengths, std::vector< ck::index_t > &input_ncdhw_stride, std::vector< ck::index_t > &output_ncdhw_stride, std::vector< ck::index_t > &, std::vector< ck::index_t > &window_spatial_zyx_lengths, std::vector< ck::index_t > &window_zyx_strides, std::vector< ck::index_t > &window_zyx_dilations, std::vector< ck::index_t > &input_left_dhw_pads, std::vector< ck::index_t > &input_right_dhw_pads)
Public Member Functions inherited from ck::tensor_operation::device::BaseArgument
 BaseArgument ()=default
 BaseArgument (const BaseArgument &)=default
BaseArgument & operator= (const BaseArgument &)=default
virtual ~BaseArgument ()

Public Attributes

const InDataType * p_in_dev_
OutDataType * p_out_dev_
IndexDataType * p_out_indices_dev_
AGridDesc_M_K a_grid_desc_m_k_
BGridDesc_M b_grid_desc_m_
InElementwiseOperation in_element_op_
AccElementwiseOperation acc_element_op_
std::vector< ck::index_t > input_ncdhw_lengths_
std::vector< ck::index_t > output_ncdhw_lengths_
std::vector< ck::index_t > input_ncdhw_stride_
std::vector< ck::index_t > output_ncdhw_stride_
Public Attributes inherited from ck::tensor_operation::device::BaseArgument
void * p_workspace_ = nullptr

Constructor & Destructor Documentation

◆ Argument()

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::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::Argument ( const InDataType * p_in_dev,
OutDataType * p_out_dev,
IndexDataType * p_out_indices_dev,
std::vector< ck::index_t > & input_ncdhw_lengths,
std::vector< ck::index_t > & output_ncdhw_lengths,
std::vector< ck::index_t > & input_ncdhw_stride,
std::vector< ck::index_t > & output_ncdhw_stride,
std::vector< ck::index_t > & ,
std::vector< ck::index_t > & window_spatial_zyx_lengths,
std::vector< ck::index_t > & window_zyx_strides,
std::vector< ck::index_t > & window_zyx_dilations,
std::vector< ck::index_t > & input_left_dhw_pads,
std::vector< ck::index_t > & input_right_dhw_pads )
inline

Member Data Documentation

◆ a_grid_desc_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>
AGridDesc_M_K ck::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::a_grid_desc_m_k_

◆ acc_element_op_

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>
AccElementwiseOperation ck::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::acc_element_op_

◆ b_grid_desc_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>
BGridDesc_M ck::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::b_grid_desc_m_

◆ in_element_op_

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>
InElementwiseOperation ck::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::in_element_op_

◆ input_ncdhw_lengths_

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::vector<ck::index_t> ck::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::input_ncdhw_lengths_

◆ input_ncdhw_stride_

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::vector<ck::index_t> ck::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::input_ncdhw_stride_

◆ output_ncdhw_lengths_

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::vector<ck::index_t> ck::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::output_ncdhw_lengths_

◆ output_ncdhw_stride_

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::vector<ck::index_t> ck::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::output_ncdhw_stride_

◆ p_in_dev_

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>
const InDataType* ck::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::p_in_dev_

◆ p_out_dev_

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>
OutDataType* ck::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::p_out_dev_

◆ p_out_indices_dev_

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>
IndexDataType* ck::tensor_operation::device::DevicePool3dFwd_NDHWC_NDHWC< InDataType, OutDataType, IndexDataType, ComputeDataType, ReduceOpId, OutputIndex, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::p_out_indices_dev_
The documentation for this struct was generated from the following file: