public static class FusedBatchNormGradV3.Options extends Object
FusedBatchNormGradV3
Modifier and Type | Method and Description |
---|---|
FusedBatchNormGradV3.Options |
dataFormat(String dataFormat) |
FusedBatchNormGradV3.Options |
epsilon(Float epsilon) |
FusedBatchNormGradV3.Options |
isTraining(Boolean isTraining) |
public FusedBatchNormGradV3.Options epsilon(Float epsilon)
epsilon
- A small float number added to the variance of x.public FusedBatchNormGradV3.Options dataFormat(String dataFormat)
dataFormat
- The data format for y_backprop, x, x_backprop.
Either "NHWC" (default) or "NCHW".public FusedBatchNormGradV3.Options isTraining(Boolean isTraining)
isTraining
- A bool value to indicate the operation is for training (default)
or inference.Copyright © 2022. All rights reserved.