org.tensorflow.op.core

Class FusedBatchNormGradV3.Options

java.lang.Object

org.tensorflow.op.core.FusedBatchNormGradV3.Options

Enclosing class:

FusedBatchNormGradV3<T extends Number,U extends Number>

public static class FusedBatchNormGradV3.Options
extends Object

Optional attributes for FusedBatchNormGradV3

Method Summary

Modifier and Type	Method and Description
FusedBatchNormGradV3.Options	dataFormat(String dataFormat)
FusedBatchNormGradV3.Options	epsilon(Float epsilon)
FusedBatchNormGradV3.Options	isTraining(Boolean isTraining)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

epsilon

public FusedBatchNormGradV3.Options epsilon(Float epsilon)

Parameters:

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

dataFormat

public FusedBatchNormGradV3.Options dataFormat(String dataFormat)

Parameters:

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

isTraining

public FusedBatchNormGradV3.Options isTraining(Boolean isTraining)

Parameters:

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