public class cudnnBatchNormMode extends Object
Modifier and Type | Field and Description |
---|---|
static int |
CUDNN_BATCHNORM_PER_ACTIVATION
bnScale, bnBias tensor dims are 1xCxHxWx..
|
static int |
CUDNN_BATCHNORM_SPATIAL
bnScale, bnBias tensor dims are 1xCx1x1 (one value per C-dim normalized over Nx1xHxW subtensors)
|
static int |
CUDNN_BATCHNORM_SPATIAL_PERSISTENT
bnScale, bnBias tensor dims are 1xCx1x1 (one value per C-dim normalized over Nx1xHxW subtensors).
|
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_PER_ACTIVATION
public static final int CUDNN_BATCHNORM_SPATIAL
public static final int CUDNN_BATCHNORM_SPATIAL_PERSISTENT
bnScale, bnBias tensor dims are 1xCx1x1 (one value per C-dim normalized over Nx1xHxW subtensors). May be faster than CUDNN_BATCHNORM_SPATIAL but imposes some limits on the range of values
public static String stringFor(int n)
Copyright © 2018. All rights reserved.