public static class
FusedBatchNormGrad.Options
Optional attributes for FusedBatchNormGrad
Public Methods
FusedBatchNormGrad.Options |
dataFormat(String dataFormat)
|
FusedBatchNormGrad.Options |
epsilon(Float epsilon)
|
FusedBatchNormGrad.Options |
isTraining(Boolean isTraining)
|
Inherited Methods
Public Methods
public FusedBatchNormGrad.Options dataFormat (String dataFormat)
Parameters
dataFormat | The data format for y_backprop, x, x_backprop. Either "NHWC" (default) or "NCHW". |
---|
public FusedBatchNormGrad.Options epsilon (Float epsilon)
Parameters
epsilon | A small float number added to the variance of x. |
---|
public FusedBatchNormGrad.Options isTraining (Boolean isTraining)
Parameters
isTraining | A bool value to indicate the operation is for training (default) or inference. |
---|