public class cudnnMultiHeadAttnWeightKind extends Object
Modifier and Type | Field and Description |
---|---|
static int |
CUDNN_MH_ATTN_K_BIASES
input projection bias for 'keys'
|
static int |
CUDNN_MH_ATTN_K_WEIGHTS
input projection weights for 'keys'
|
static int |
CUDNN_MH_ATTN_O_BIASES
output projection biases
|
static int |
CUDNN_MH_ATTN_O_WEIGHTS
output projection weights
|
static int |
CUDNN_MH_ATTN_Q_BIASES
input projection bias tensor for 'queries'
|
static int |
CUDNN_MH_ATTN_Q_WEIGHTS
input projection weights for 'queries'
|
static int |
CUDNN_MH_ATTN_V_BIASES
input projection bias for 'values'
|
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 final int CUDNN_MH_ATTN_Q_BIASES
public static final int CUDNN_MH_ATTN_K_BIASES
public static final int CUDNN_MH_ATTN_V_BIASES
public static final int CUDNN_MH_ATTN_O_BIASES
public static String stringFor(int n)
Copyright © 2020. All rights reserved.