#include <training_ops.h>
Optional attribute setters for SparseApplyMomentum.
Public attributes | |
---|---|
use_locking_ = false | bool |
use_nesterov_ = false | bool |
Public functions | |
---|---|
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. |
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. |
bool tensorflow::ops::SparseApplyMomentum::Attrs::use_locking_ = false
bool tensorflow::ops::SparseApplyMomentum::Attrs::use_nesterov_ = false
Attrs tensorflow::ops::SparseApplyMomentum::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
Attrs tensorflow::ops::SparseApplyMomentum::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
© 2017 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/cc/struct/tensorflow/ops/sparse-apply-momentum/attrs.html