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

From class java.lang.Object

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.