public class cudnnBatchNormOps extends Object
Modifier and Type | Field and Description |
---|---|
static int |
CUDNN_BATCHNORM_OPS_BN
do batch normalization only
|
static int |
CUDNN_BATCHNORM_OPS_BN_ACTIVATION
do batchNorm, then activation
|
static int |
CUDNN_BATCHNORM_OPS_BN_ADD_ACTIVATION
do batchNorm, then elemWiseAdd, then activation
|
Modifier and Type | Method and Description |
---|---|
static String |
stringFor(int n)
Returns a string representation of the given constant
|
public static final int CUDNN_BATCHNORM_OPS_BN
public static final int CUDNN_BATCHNORM_OPS_BN_ACTIVATION
public static final int CUDNN_BATCHNORM_OPS_BN_ADD_ACTIVATION
public static String stringFor(int n)
Copyright © 2020. All rights reserved.