TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
Public Attributes | List of all members
bert::TFusedMultiHeadAttentionXMMAKernel< TKernelMeta, TKernelParam >::FusedMultiHeadAttentionKernelInfo Struct Reference
Collaboration diagram for bert::TFusedMultiHeadAttentionXMMAKernel< TKernelMeta, TKernelParam >::FusedMultiHeadAttentionKernelInfo:
Collaboration graph
[legend]

Public Attributes

unsigned int mMetaInfoIndex
 
CUfunction mDeviceFunction
 

Member Data Documentation

◆ mMetaInfoIndex

template<typename TKernelMeta , typename TKernelParam >
unsigned int bert::TFusedMultiHeadAttentionXMMAKernel< TKernelMeta, TKernelParam >::FusedMultiHeadAttentionKernelInfo::mMetaInfoIndex

◆ mDeviceFunction

template<typename TKernelMeta , typename TKernelParam >
CUfunction bert::TFusedMultiHeadAttentionXMMAKernel< TKernelMeta, TKernelParam >::FusedMultiHeadAttentionKernelInfo::mDeviceFunction
The documentation for this struct was generated from the following file: