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