tensorflow:: ops:: ApplyMomentum:: Attrs
#include <training_ops.h>
Optional attribute setters for ApplyMomentum.
Summary
Public attributes |
|
---|---|
use_locking_ = false
|
bool
|
use_nesterov_ = false
|
bool
|
Public functions |
|
---|---|
UseLocking(bool x)
|
TF_MUST_USE_RESULT Attrs
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(bool x)
|
TF_MUST_USE_RESULT Attrs
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. |
Public attributes
use_locking_
bool tensorflow::ops::ApplyMomentum::Attrs::use_locking_ = false
use_nesterov_
bool tensorflow::ops::ApplyMomentum::Attrs::use_nesterov_ = false
Public functions
UseLocking
TF_MUST_USE_RESULT Attrs tensorflow::ops::ApplyMomentum::Attrs::UseLocking( bool x )
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.
Defaults to false
UseNesterov
TF_MUST_USE_RESULT Attrs tensorflow::ops::ApplyMomentum::Attrs::UseNesterov( bool x )
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.
Defaults to false