ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

ResourceApplyKerasMomentum.Options

public static class ResourceApplyKerasMomentum.Options

Optional attributes for ResourceApplyKerasMomentum

Public Methods

ResourceApplyKerasMomentum.Options
useLocking(Boolean useLocking)
ResourceApplyKerasMomentum.Options
useNesterov(Boolean useNesterov)

Inherited Methods

Public Methods

public ResourceApplyKerasMomentum.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.

public ResourceApplyKerasMomentum.Options useNesterov (Boolean useNesterov)

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