public static class ResourceApplyKerasMomentum.Options extends Object
ResourceApplyKerasMomentum
Modifier and Type | Method and Description |
---|---|
ResourceApplyKerasMomentum.Options |
useLocking(Boolean useLocking) |
ResourceApplyKerasMomentum.Options |
useNesterov(Boolean useNesterov) |
public ResourceApplyKerasMomentum.Options useLocking(Boolean useLocking)
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)
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.Copyright © 2022. All rights reserved.