Searched refs:set_use_locking (Results 1 – 4 of 4) sorted by relevance
30 this->set_use_locking(use_locking); in Init()38 void ApplyMomentum::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::ApplyMomentum
56 this->set_use_locking(use_locking); in Init()60 void Adam::set_use_locking(const bool use_locking) { (void)this->AddAttr(kUseLocking, MakeValue(use… in set_use_locking() function in mindspore::ops::Adam
42 void set_use_locking(const bool use_locking);
45 void set_use_locking(const bool use_locking);