public class cudnnMultiHeadAttnWeightKind extends Object
Modifier and Type | Field and Description |
---|---|
static int |
CUDNN_MH_ATTN_K_WEIGHTS
input projection weights for 'keys'
|
static int |
CUDNN_MH_ATTN_O_WEIGHTS
output projection weights
|
static int |
CUDNN_MH_ATTN_Q_WEIGHTS
input projection weights for 'queries'
|
static int |
CUDNN_MH_ATTN_V_WEIGHTS
input projection weights for 'values'
|
Modifier and Type | Method and Description |
---|---|
static String |
stringFor(int n)
Returns a string representation of the given constant
|
public static final int CUDNN_MH_ATTN_Q_WEIGHTS
public static final int CUDNN_MH_ATTN_K_WEIGHTS
public static final int CUDNN_MH_ATTN_V_WEIGHTS
public static final int CUDNN_MH_ATTN_O_WEIGHTS
public static String stringFor(int n)
Copyright © 2019. All rights reserved.