Argument Struct Reference

Argument Struct Reference#

Composable Kernel: ck::tensor_operation::device::DeviceMaxPoolBwdImpl< DOutDataType, IndexDataType, DInDataType, InOutVectorSize >::Argument Struct Reference

#include <device_max_pool_bwd_impl.hpp>

Inheritance diagram for ck::tensor_operation::device::DeviceMaxPoolBwdImpl< DOutDataType, IndexDataType, DInDataType, InOutVectorSize >::Argument:

Public Member Functions
	Argument (const DOutDataType p_dout, const IndexDataType p_indices, DInDataType *p_din, index_t dout_length, index_t din_length, const std::vector< ck::index_t > &window_lengths, const std::vector< ck::index_t > &window_strides, const std::vector< ck::index_t > &window_dilations)
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_
const IndexDataType *	p_indices_
DInDataType *	p_din_
index_t	dout_length_raw_
index_t	din_length_raw_
index_t	blockSize_
bool	windowOverlap_
Public Attributes inherited from ck::tensor_operation::device::BaseArgument
void *	p_workspace_ = nullptr

Constructor & Destructor Documentation

◆ Argument()

template<typename DOutDataType, typename IndexDataType, typename DInDataType, ck::index_t InOutVectorSize>

ck::tensor_operation::device::DeviceMaxPoolBwdImpl< DOutDataType, IndexDataType, DInDataType, InOutVectorSize >::Argument::Argument	(	const DOutDataType *	p_dout,
		const IndexDataType *	p_indices,
		DInDataType *	p_din,
		index_t	dout_length,
		index_t	din_length,
		const std::vector< ck::index_t > &	window_lengths,
		const std::vector< ck::index_t > &	window_strides,
		const std::vector< ck::index_t > &	window_dilations )

inline

Member Data Documentation

◆ blockSize_

template<typename DOutDataType, typename IndexDataType, typename DInDataType, ck::index_t InOutVectorSize>

index_t ck::tensor_operation::device::DeviceMaxPoolBwdImpl< DOutDataType, IndexDataType, DInDataType, InOutVectorSize >::Argument::blockSize_

◆ din_length_raw_

template<typename DOutDataType, typename IndexDataType, typename DInDataType, ck::index_t InOutVectorSize>

index_t ck::tensor_operation::device::DeviceMaxPoolBwdImpl< DOutDataType, IndexDataType, DInDataType, InOutVectorSize >::Argument::din_length_raw_

◆ dout_length_raw_

template<typename DOutDataType, typename IndexDataType, typename DInDataType, ck::index_t InOutVectorSize>

index_t ck::tensor_operation::device::DeviceMaxPoolBwdImpl< DOutDataType, IndexDataType, DInDataType, InOutVectorSize >::Argument::dout_length_raw_

◆ p_din_

template<typename DOutDataType, typename IndexDataType, typename DInDataType, ck::index_t InOutVectorSize>

DInDataType* ck::tensor_operation::device::DeviceMaxPoolBwdImpl< DOutDataType, IndexDataType, DInDataType, InOutVectorSize >::Argument::p_din_

◆ p_dout_

template<typename DOutDataType, typename IndexDataType, typename DInDataType, ck::index_t InOutVectorSize>

const DOutDataType* ck::tensor_operation::device::DeviceMaxPoolBwdImpl< DOutDataType, IndexDataType, DInDataType, InOutVectorSize >::Argument::p_dout_

◆ p_indices_

template<typename DOutDataType, typename IndexDataType, typename DInDataType, ck::index_t InOutVectorSize>

const IndexDataType* ck::tensor_operation::device::DeviceMaxPoolBwdImpl< DOutDataType, IndexDataType, DInDataType, InOutVectorSize >::Argument::p_indices_

◆ windowOverlap_

template<typename DOutDataType, typename IndexDataType, typename DInDataType, ck::index_t InOutVectorSize>

bool ck::tensor_operation::device::DeviceMaxPoolBwdImpl< DOutDataType, IndexDataType, DInDataType, InOutVectorSize >::Argument::windowOverlap_

The documentation for this struct was generated from the following file:

device_max_pool_bwd_impl.hpp