jcuda.jcudnn

Class cudnnFusedOps

java.lang.Object

jcuda.jcudnn.cudnnFusedOps

public class cudnnFusedOps
extends Object

Field Summary

Fields

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

Method Summary

Modifier and Type	Method and Description
static String	stringFor(int n) Returns a string representation of the given constant

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

CUDNN_FUSED_SCALE_BIAS_ACTIVATION_CONV_BNSTATS

public static final int CUDNN_FUSED_SCALE_BIAS_ACTIVATION_CONV_BNSTATS

[per channel scale], [per channel bias], [activation], convolution, [generate BN stats]

See Also:

Constant Field Values

CUDNN_FUSED_SCALE_BIAS_ACTIVATION_WGRAD

public static final int CUDNN_FUSED_SCALE_BIAS_ACTIVATION_WGRAD

[per channel scale], [per channel bias], [activation], convolutionBackwardWeights

See Also:

Constant Field Values

CUDNN_FUSED_BN_FINALIZE_STATISTICS_TRAINING

public static final int CUDNN_FUSED_BN_FINALIZE_STATISTICS_TRAINING

optionally update running stats and generate saved stats

See Also:

Constant Field Values

CUDNN_FUSED_BN_FINALIZE_STATISTICS_INFERENCE

public static final int CUDNN_FUSED_BN_FINALIZE_STATISTICS_INFERENCE

computes the equivalent scale and bias from learned running stats and learned scale, bias

See Also:

Constant Field Values

CUDNN_FUSED_CONV_SCALE_BIAS_ADD_ACTIVATION

public static final int CUDNN_FUSED_CONV_SCALE_BIAS_ADD_ACTIVATION

reserved for future use: convolution, [per channel scale], [per channel bias], [residual add], [activation]

See Also:

Constant Field Values

CUDNN_FUSED_SCALE_BIAS_ADD_ACTIVATION_GEN_BITMASK

public static final int CUDNN_FUSED_SCALE_BIAS_ADD_ACTIVATION_GEN_BITMASK

reserved for future use: [per channel scale], [per channel bias], [residual add], activation, bitmask

See Also:

Constant Field Values

CUDNN_FUSED_DACTIVATION_FORK_DBATCHNORM

public static final int CUDNN_FUSED_DACTIVATION_FORK_DBATCHNORM

reserved for future use

See Also:

Constant Field Values

Method Detail

stringFor

public static String stringFor(int n)

Returns a string representation of the given constant

Returns:

A string representation of the given constant