Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tensorflow::ops::ApplyRMSProp::Attrs

#include <training_ops.h>

Optional attribute setters for ApplyRMSProp.

Summary

Public attributes

use_locking_ = false
bool

Public functions

UseLocking(bool x)
TF_MUST_USE_RESULT Attrs
If True, updating of the var, ms, and mom tensors is protected by a lock; otherwise the behavior is undefined, but may exhibit less contention.

Public attributes

use_locking_

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

Public functions

UseLocking

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

If True, updating of the var, ms, and mom tensors is protected by a lock; otherwise the behavior is undefined, but may exhibit less contention.

Defaults to false