Class ResourceApplyMomentum.Options

java.lang.Object

org.tensorflow.op.train.ResourceApplyMomentum.Options

Enclosing class:

ResourceApplyMomentum

public static class ResourceApplyMomentum.Options
extends Object

Optional attributes for ResourceApplyMomentum

Method Summary

Modifier and Type	Method	Description
ResourceApplyMomentum.Options	useLocking(Boolean useLocking)	Sets the useLocking option.
ResourceApplyMomentum.Options	useNesterov(Boolean useNesterov)	Sets the useNesterov option.

Methods inherited from class Object

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

Method Details

useLocking

public ResourceApplyMomentum.Options useLocking(Boolean useLocking)

Sets the useLocking option.

Parameters:

useLocking - If True, updating of the var and accum tensors will be protected by a lock; otherwise the behavior is undefined, but may exhibit less contention.

Returns:

this Options instance.

useNesterov

public ResourceApplyMomentum.Options useNesterov(Boolean useNesterov)

Sets the useNesterov option.

Parameters:

useNesterov - If True, the tensor passed to compute grad will be var - lr * momentum * accum, so in the end, the var you get is actually var - lr * momentum * accum.

Returns:

this Options instance.