FusedBatchNorm.Options

public static class FusedBatchNorm.Options

Optional attributes for FusedBatchNorm

Public Methods

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

Inherited Methods

boolean
equals(Object arg0)
final Class<?>
getClass()
int
hashCode()
final void
notify()
final void
notifyAll()
String
toString()
final void
wait(long arg0, int arg1)
final void
wait(long arg0)
final void
wait()

Public Methods

public FusedBatchNorm.Options dataFormat (String dataFormat)

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

public FusedBatchNorm.Options epsilon (Float epsilon)

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

public FusedBatchNorm.Options exponentialAvgFactor (Float exponentialAvgFactor)

public FusedBatchNorm.Options isTraining (Boolean isTraining)

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