FusedBatchNormGrad.Options

public static class FusedBatchNormGrad.Options

Optional attributes for FusedBatchNormGrad

Public Methods

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 FusedBatchNormGrad.Options dataFormat (String dataFormat)

Parameters
dataFormat The data format for y_backprop, x, x_backprop. Either "NHWC" (default) or "NCHW".

public FusedBatchNormGrad.Options epsilon (Float epsilon)

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

public FusedBatchNormGrad.Options isTraining (Boolean isTraining)

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