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

tensorflow:: ops:: ApplyAdam:: Attrs

#include <training_ops.h>

Optional attribute setters for ApplyAdam .

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, m, and v 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 , uses the nesterov update.

Public attributes

use_locking_

bool tensorflow::ops::ApplyAdam::Attrs::use_locking_ = false

use_nesterov_

bool tensorflow::ops::ApplyAdam::Attrs::use_nesterov_ = false

Public functions

UseLocking

TF_MUST_USE_RESULT Attrs tensorflow::ops::ApplyAdam::Attrs::UseLocking(
  bool x
)

If True , updating of the var, m, and v 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::ApplyAdam::Attrs::UseNesterov(
  bool x
)

If True , uses the nesterov update.

Defaults to false