org.tensorflow.op.train

Class ResourceSparseApplyMomentum.Options

java.lang.Object

org.tensorflow.op.train.ResourceSparseApplyMomentum.Options

Enclosing class:

ResourceSparseApplyMomentum

public static class ResourceSparseApplyMomentum.Options
extends Object

Optional attributes for ResourceSparseApplyMomentum

Method Summary

Modifier and Type	Method and Description
ResourceSparseApplyMomentum.Options	useLocking(Boolean useLocking)
ResourceSparseApplyMomentum.Options	useNesterov(Boolean useNesterov)

Methods inherited from class java.lang.Object

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

Method Detail

useLocking

public ResourceSparseApplyMomentum.Options useLocking(Boolean useLocking)

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.

useNesterov

public ResourceSparseApplyMomentum.Options useNesterov(Boolean useNesterov)

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.