org.tensorflow.op.core

Class FusedBatchNormGradV2.Options

java.lang.Object

org.tensorflow.op.core.FusedBatchNormGradV2.Options

Enclosing class:

FusedBatchNormGradV2<T extends Number,U extends Number>

public static class FusedBatchNormGradV2.Options
extends Object

Optional attributes for FusedBatchNormGradV2

Method Summary

Modifier and Type	Method and Description
FusedBatchNormGradV2.Options	dataFormat(String dataFormat)
FusedBatchNormGradV2.Options	epsilon(Float epsilon)
FusedBatchNormGradV2.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 FusedBatchNormGradV2.Options epsilon(Float epsilon)

Parameters:

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

dataFormat

public FusedBatchNormGradV2.Options dataFormat(String dataFormat)

Parameters:

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

isTraining

public FusedBatchNormGradV2.Options isTraining(Boolean isTraining)

Parameters:

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