device_batched_gemm_softmax_gemm.hpp File Reference

• include
• ck
• tensor_operation
• gpu
• device

Classes | Namespaces

device_batched_gemm_softmax_gemm.hpp File Reference

#include <iostream>
#include <vector>
#include "device_base.hpp"

Go to the source code of this file.

Classes
struct	ck::tensor_operation::device::DeviceBatchedGemmSoftmaxGemm< ALayout, B0Layout, B1Layout, CLayout, ADataType, B0DataType, B1DataType, CDataType, AElementwiseOperation, B0ElementwiseOperation, Acc0ElementwiseOperation, B1ElementwiseOperation, CElementwiseOperation, MaskOutUpperTriangle >

Namespaces
namespace	ck
namespace	ck::tensor_operation
namespace	ck::tensor_operation::device