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. |
|---|