Searched defs:set_use_locking (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ | ||
| D | apply_momentum.cc | 38 void ApplyMomentum::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::ApplyMomentum |
| D | adam.cc | 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 |