public class cudnnFusedOps extends Object
Modifier and Type | Field and Description |
---|---|
static int |
CUDNN_FUSED_BN_FINALIZE_STATISTICS_INFERENCE
computes the equivalent scale and bias from learned running stats and learned scale, bias
|
static int |
CUDNN_FUSED_BN_FINALIZE_STATISTICS_TRAINING
optionally update running stats and generate saved stats
|
static int |
CUDNN_FUSED_CONV_SCALE_BIAS_ADD_ACTIVATION
reserved for future use: convolution, [per channel scale], [per channel bias], [residual add], [activation]
|
static int |
CUDNN_FUSED_DACTIVATION_FORK_DBATCHNORM
reserved for future use
|
static int |
CUDNN_FUSED_SCALE_BIAS_ACTIVATION_CONV_BNSTATS
[per channel scale], [per channel bias], [activation], convolution, [generate BN stats]
|
static int |
CUDNN_FUSED_SCALE_BIAS_ACTIVATION_WGRAD
[per channel scale], [per channel bias], [activation], convolutionBackwardWeights
|
static int |
CUDNN_FUSED_SCALE_BIAS_ADD_ACTIVATION_GEN_BITMASK
reserved for future use: [per channel scale], [per channel bias], [residual add], activation, bitmask
|
Modifier and Type | Method and Description |
---|---|
static String |
stringFor(int n)
Returns a string representation of the given constant
|
public static final int CUDNN_FUSED_SCALE_BIAS_ACTIVATION_CONV_BNSTATS
public static final int CUDNN_FUSED_SCALE_BIAS_ACTIVATION_WGRAD
public static final int CUDNN_FUSED_BN_FINALIZE_STATISTICS_TRAINING
public static final int CUDNN_FUSED_BN_FINALIZE_STATISTICS_INFERENCE
public static final int CUDNN_FUSED_CONV_SCALE_BIAS_ADD_ACTIVATION
public static final int CUDNN_FUSED_SCALE_BIAS_ADD_ACTIVATION_GEN_BITMASK
public static final int CUDNN_FUSED_DACTIVATION_FORK_DBATCHNORM
public static String stringFor(int n)
Copyright © 2019. All rights reserved.