Member List

• ck
• tensor_operation
• device
• DeviceSparseEmbeddingsForwardLayernorm

ck::tensor_operation::device::DeviceSparseEmbeddingsForwardLayernorm< EmbType, IndexType, GammaDataType, BetaDataType, AccDataType, OutType, EmbElementwiseOperation, BlockSize, DimClusterSize, RowClusterSize, DimPerBlock, RowPerBlock, DimThreadSize, RowVectorSize, NumEmbeddings > Member List

This is the complete list of members for ck::tensor_operation::device::DeviceSparseEmbeddingsForwardLayernorm< EmbType, IndexType, GammaDataType, BetaDataType, AccDataType, OutType, EmbElementwiseOperation, BlockSize, DimClusterSize, RowClusterSize, DimPerBlock, RowPerBlock, DimThreadSize, RowVectorSize, NumEmbeddings >, including all inherited members.

BaseOperator()=default	ck::tensor_operation::device::BaseOperator
BaseOperator(const BaseOperator &)=default	ck::tensor_operation::device::BaseOperator
GetInstanceString() const	ck::tensor_operation::device::BaseOperator	inlinevirtual
GetObjectName() const	ck::tensor_operation::device::BaseOperator	inlinevirtual
GetTemplateInfo() const	ck::tensor_operation::device::BaseOperator	inlinevirtual
GetTypeIdHashCode() const	ck::tensor_operation::device::BaseOperator	inlinevirtual
GetTypeIdName() const	ck::tensor_operation::device::BaseOperator	inlinevirtual
GetTypeString() const override	ck::tensor_operation::device::DeviceSparseEmbeddingsForwardLayernorm< EmbType, IndexType, GammaDataType, BetaDataType, AccDataType, OutType, EmbElementwiseOperation, BlockSize, DimClusterSize, RowClusterSize, DimPerBlock, RowPerBlock, DimThreadSize, RowVectorSize, NumEmbeddings >	inlinevirtual
GetWorkSpaceSize(const BaseArgument *) const	ck::tensor_operation::device::BaseOperator	inlinevirtual
GridwiseSparseEmbedding typedef	ck::tensor_operation::device::DeviceSparseEmbeddingsForwardLayernorm< EmbType, IndexType, GammaDataType, BetaDataType, AccDataType, OutType, EmbElementwiseOperation, BlockSize, DimClusterSize, RowClusterSize, DimPerBlock, RowPerBlock, DimThreadSize, RowVectorSize, NumEmbeddings >
IsSupportedArgument(const Argument *p_arg)	ck::tensor_operation::device::DeviceSparseEmbeddingsForwardLayernorm< EmbType, IndexType, GammaDataType, BetaDataType, AccDataType, OutType, EmbElementwiseOperation, BlockSize, DimClusterSize, RowClusterSize, DimPerBlock, RowPerBlock, DimThreadSize, RowVectorSize, NumEmbeddings >	inlinestatic
IsSupportedArgument(const BaseArgument *p_arg) override	ck::tensor_operation::device::DeviceSparseEmbeddingsForwardLayernorm< EmbType, IndexType, GammaDataType, BetaDataType, AccDataType, OutType, EmbElementwiseOperation, BlockSize, DimClusterSize, RowClusterSize, DimPerBlock, RowPerBlock, DimThreadSize, RowVectorSize, NumEmbeddings >	inlinevirtual
MakeArgumentPointer(void *p_out, const ck::Array< EmbType *, NumEmbeddings > &p_embs, const ck::Array< IndexType *, NumEmbeddings > &p_indexs, const void *p_gamma, const void *p_beta, ck::index_t EmbeddingDim, ck::index_t IndexLength, const AccDataType epsilon, const EmbElementwiseOperation emb_elementwise_op)	ck::tensor_operation::device::DeviceSparseEmbeddingsForwardLayernorm< EmbType, IndexType, GammaDataType, BetaDataType, AccDataType, OutType, EmbElementwiseOperation, BlockSize, DimClusterSize, RowClusterSize, DimPerBlock, RowPerBlock, DimThreadSize, RowVectorSize, NumEmbeddings >	inline
MakeInvokerPointer()	ck::tensor_operation::device::DeviceSparseEmbeddingsForwardLayernorm< EmbType, IndexType, GammaDataType, BetaDataType, AccDataType, OutType, EmbElementwiseOperation, BlockSize, DimClusterSize, RowClusterSize, DimPerBlock, RowPerBlock, DimThreadSize, RowVectorSize, NumEmbeddings >	inlinevirtual
MakeOutputDescriptor(const index_t index_length, const index_t rows)	ck::tensor_operation::device::DeviceSparseEmbeddingsForwardLayernorm< EmbType, IndexType, GammaDataType, BetaDataType, AccDataType, OutType, EmbElementwiseOperation, BlockSize, DimClusterSize, RowClusterSize, DimPerBlock, RowPerBlock, DimThreadSize, RowVectorSize, NumEmbeddings >	inlinestatic
operator=(const BaseOperator &)=default	ck::tensor_operation::device::BaseOperator
SetWorkSpacePointer(BaseArgument *p_arg, void *p_workspace, const StreamConfig &=StreamConfig{}) const	ck::tensor_operation::device::BaseOperator	inlinevirtual
~BaseOperator()	ck::tensor_operation::device::BaseOperator	inlinevirtual