ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

FusedBatchNormV3.Options

public static class FusedBatchNormV3.Options

Optional attributes for FusedBatchNormV3

Public Methods

FusedBatchNormV3.Options
dataFormat(String dataFormat)
FusedBatchNormV3.Options
epsilon(Float epsilon)
FusedBatchNormV3.Options
exponentialAvgFactor(Float exponentialAvgFactor)
FusedBatchNormV3.Options
isTraining(Boolean isTraining)

Inherited Methods

Public Methods

public FusedBatchNormV3.Options dataFormat (String dataFormat)

Parameters
dataFormat The data format for x and y. Either "NHWC" (default) or "NCHW".

public FusedBatchNormV3.Options epsilon (Float epsilon)

Parameters
epsilon A small float number added to the variance of x.

public FusedBatchNormV3.Options exponentialAvgFactor (Float exponentialAvgFactor)

public FusedBatchNormV3.Options isTraining (Boolean isTraining)

Parameters
isTraining A bool value to indicate the operation is for training (default) or inference.