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