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