DeviceElementwiseNormalization< InDataTypeTuple, GammaDataType, BetaDataType, AccDataType, YDataType, XElementwiseOperation, YElementwiseOperation, Rank, NumReduceDim > Struct Template Reference

• ck
• tensor_operation
• device
• DeviceElementwiseNormalization

Public Member Functions | Static Public Attributes | List of all members

ck::tensor_operation::device::DeviceElementwiseNormalization< InDataTypeTuple, GammaDataType, BetaDataType, AccDataType, YDataType, XElementwiseOperation, YElementwiseOperation, Rank, NumReduceDim > Struct Template Referenceabstract

#include <device_elementwise_normalization.hpp>

Inheritance diagram for ck::tensor_operation::device::DeviceElementwiseNormalization< InDataTypeTuple, GammaDataType, BetaDataType, AccDataType, YDataType, XElementwiseOperation, YElementwiseOperation, Rank, NumReduceDim >:

Public Member Functions
virtual std::unique_ptr< BaseArgument >	MakeArgumentPointer (const std::vector< index_t > lengths, const std::array< std::vector< index_t >, NumInput > inStridesArray, const std::vector< index_t > gammaStrides, const std::vector< index_t > betaStrides, const std::vector< index_t > yStrides, const std::vector< index_t > reduceDims, double epsilon, const std::array< const void *, NumInput > in_dev_buffers, const void *p_gamma, const void *p_beta, void *p_y, XElementwiseOperation x_elementwise_op, YElementwiseOperation y_elementwise_op)=0
virtual std::unique_ptr< BaseInvoker >	MakeInvokerPointer ()=0
Public Member Functions inherited from ck::tensor_operation::device::BaseOperator
	BaseOperator ()=default
	BaseOperator (const BaseOperator &)=default
BaseOperator &	operator= (const BaseOperator &)=default
virtual bool	IsSupportedArgument (const BaseArgument *)
virtual std::string	GetTypeString () const
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 Attributes
static constexpr int	NumInput = InDataTypeTuple::Size()

Member Function Documentation

◆ MakeArgumentPointer()

template<typename InDataTypeTuple, typename GammaDataType, typename BetaDataType, typename AccDataType, typename YDataType, typename XElementwiseOperation, typename YElementwiseOperation, index_t Rank, index_t NumReduceDim>

virtual std::unique_ptr< BaseArgument > ck::tensor_operation::device::DeviceElementwiseNormalization< InDataTypeTuple, GammaDataType, BetaDataType, AccDataType, YDataType, XElementwiseOperation, YElementwiseOperation, Rank, NumReduceDim >::MakeArgumentPointer ( const std::vector< index_t > lengths, const std::array< std::vector< index_t >, NumInput > inStridesArray, const std::vector< index_t > gammaStrides, const std::vector< index_t > betaStrides, const std::vector< index_t > yStrides, const std::vector< index_t > reduceDims, double epsilon, const std::array< const void *, NumInput > in_dev_buffers, const void * p_gamma, const void * p_beta, void * p_y, XElementwiseOperation x_elementwise_op, YElementwiseOperation y_elementwise_op )

pure virtual

Implemented in ck::tensor_operation::device::DeviceElementwiseNormalizationImpl< InDataTypeTuple, GammaDataType, BetaDataType, AccDataType, YDataType, XElementwiseOperation, YElementwiseOperation, Rank, NumReduceDim, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, XYSrcVectorDim, XSrcVectorSize, GammaSrcVectorDim, GammaSrcVectorSize, BetaSrcVectorDim, BetaSrcVectorSize, YDstVectorSize >.

◆ MakeInvokerPointer()

template<typename InDataTypeTuple, typename GammaDataType, typename BetaDataType, typename AccDataType, typename YDataType, typename XElementwiseOperation, typename YElementwiseOperation, index_t Rank, index_t NumReduceDim>

virtual std::unique_ptr< BaseInvoker > ck::tensor_operation::device::DeviceElementwiseNormalization< InDataTypeTuple, GammaDataType, BetaDataType, AccDataType, YDataType, XElementwiseOperation, YElementwiseOperation, Rank, NumReduceDim >::MakeInvokerPointer ( )

pure virtual

Implemented in ck::tensor_operation::device::DeviceElementwiseNormalizationImpl< InDataTypeTuple, GammaDataType, BetaDataType, AccDataType, YDataType, XElementwiseOperation, YElementwiseOperation, Rank, NumReduceDim, BlockSize, MThreadClusterSize, KThreadClusterSize, MThreadSliceSize, KThreadSliceSize, XYSrcVectorDim, XSrcVectorSize, GammaSrcVectorDim, GammaSrcVectorSize, BetaSrcVectorDim, BetaSrcVectorSize, YDstVectorSize >.

Member Data Documentation

◆ NumInput

template<typename InDataTypeTuple, typename GammaDataType, typename BetaDataType, typename AccDataType, typename YDataType, typename XElementwiseOperation, typename YElementwiseOperation, index_t Rank, index_t NumReduceDim>

int ck::tensor_operation::device::DeviceElementwiseNormalization< InDataTypeTuple, GammaDataType, BetaDataType, AccDataType, YDataType, XElementwiseOperation, YElementwiseOperation, Rank, NumReduceDim >::NumInput = InDataTypeTuple::Size()

staticconstexpr

---

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

• device_elementwise_normalization.hpp