Argument Struct Reference#
ck::tensor_operation::device::DeviceAvgPool3dBwd_NDHWC_NDHWC< DOutDataType, DInDataType, ComputeDataType, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument Struct Reference
#include <device_avgpool3d_bwd_ndhwc_ndhwc.hpp>
Inheritance diagram for ck::tensor_operation::device::DeviceAvgPool3dBwd_NDHWC_NDHWC< DOutDataType, DInDataType, ComputeDataType, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument:
Public Member Functions | |
| Argument (const DOutDataType *p_dout, DInDataType *p_din, std::vector< ck::index_t > dout_n_c_wos_lengths, std::vector< ck::index_t > din_n_c_wos_length, std::vector< ck::index_t > dout_n_c_wos_strides, std::vector< ck::index_t > din_n_c_wos_strides, std::vector< ck::index_t > window_lengths, std::vector< ck::index_t > window_strides, std::vector< ck::index_t > window_dilations, std::vector< ck::index_t > input_left_pads, std::vector< ck::index_t > input_right_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 DOutDataType * | p_dout_grid_ |
| DInDataType * | p_din_grid_ |
| std::vector< ck::index_t > | dout_n_c_wos_lengths_ |
| std::vector< ck::index_t > | din_n_c_wos_length_ |
| std::vector< ck::index_t > | dout_n_c_wos_strides_ |
| std::vector< ck::index_t > | din_n_c_wos_strides_ |
| int | num_reduce_ |
| std::vector< DoutGridDesc_M_K > | dout_grid_desc_m_k_container_ |
| std::vector< DinGridDesc_M > | din_grid_desc_m_container_ |
| Div | div_element_op_ |
| Public Attributes inherited from ck::tensor_operation::device::BaseArgument | |
| void * | p_workspace_ = nullptr |
Constructor & Destructor Documentation
◆ Argument()
template<typename DOutDataType, typename DInDataType, typename ComputeDataType, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
|
inline |
Member Data Documentation
◆ din_grid_desc_m_container_
template<typename DOutDataType, typename DInDataType, typename ComputeDataType, 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<DinGridDesc_M> ck::tensor_operation::device::DeviceAvgPool3dBwd_NDHWC_NDHWC< DOutDataType, DInDataType, ComputeDataType, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::din_grid_desc_m_container_ |
◆ din_n_c_wos_length_
template<typename DOutDataType, typename DInDataType, typename ComputeDataType, 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::DeviceAvgPool3dBwd_NDHWC_NDHWC< DOutDataType, DInDataType, ComputeDataType, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::din_n_c_wos_length_ |
◆ din_n_c_wos_strides_
template<typename DOutDataType, typename DInDataType, typename ComputeDataType, 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::DeviceAvgPool3dBwd_NDHWC_NDHWC< DOutDataType, DInDataType, ComputeDataType, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::din_n_c_wos_strides_ |
◆ div_element_op_
template<typename DOutDataType, typename DInDataType, typename ComputeDataType, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
| Div ck::tensor_operation::device::DeviceAvgPool3dBwd_NDHWC_NDHWC< DOutDataType, DInDataType, ComputeDataType, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::div_element_op_ |
◆ dout_grid_desc_m_k_container_
template<typename DOutDataType, typename DInDataType, typename ComputeDataType, 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<DoutGridDesc_M_K> ck::tensor_operation::device::DeviceAvgPool3dBwd_NDHWC_NDHWC< DOutDataType, DInDataType, ComputeDataType, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::dout_grid_desc_m_k_container_ |
◆ dout_n_c_wos_lengths_
template<typename DOutDataType, typename DInDataType, typename ComputeDataType, 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::DeviceAvgPool3dBwd_NDHWC_NDHWC< DOutDataType, DInDataType, ComputeDataType, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::dout_n_c_wos_lengths_ |
◆ dout_n_c_wos_strides_
template<typename DOutDataType, typename DInDataType, typename ComputeDataType, 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::DeviceAvgPool3dBwd_NDHWC_NDHWC< DOutDataType, DInDataType, ComputeDataType, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::dout_n_c_wos_strides_ |
◆ num_reduce_
template<typename DOutDataType, typename DInDataType, typename ComputeDataType, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
| int ck::tensor_operation::device::DeviceAvgPool3dBwd_NDHWC_NDHWC< DOutDataType, DInDataType, ComputeDataType, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::num_reduce_ |
◆ p_din_grid_
template<typename DOutDataType, typename DInDataType, typename ComputeDataType, ck::index_t BlockSize, ck::index_t MThreadClusterSize, ck::index_t KThreadClusterSize, ck::index_t MThreadSliceSize, ck::index_t KThreadSliceSize, ck::index_t InSrcOutDstVectorSize>
| DInDataType* ck::tensor_operation::device::DeviceAvgPool3dBwd_NDHWC_NDHWC< DOutDataType, DInDataType, ComputeDataType, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::p_din_grid_ |
◆ p_dout_grid_
template<typename DOutDataType, typename DInDataType, typename ComputeDataType, 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 DOutDataType* ck::tensor_operation::device::DeviceAvgPool3dBwd_NDHWC_NDHWC< DOutDataType, DInDataType, ComputeDataType, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, InSrcOutDstVectorSize >::Argument::p_dout_grid_ |
The documentation for this struct was generated from the following file: