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:

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-7.0.1/include/ck/tensor_operation/gpu/device/impl/device_max_pool_bwd_impl.hpp