Home
last modified time | relevance | path

Searched defs:set_use_locking (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dscatter_nd_arithmetic.cc103 void ScatterNdUpdate::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::ScatterNdUpdate
114 void ScatterNdAdd::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::ScatterNdAdd
125 void ScatterNdSub::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::ScatterNdSub
136 void ScatterNdMul::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::ScatterNdMul
147 void ScatterNdDiv::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::ScatterNdDiv
158 void ScatterNdMax::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::ScatterNdMax
169 void ScatterNdMin::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::ScatterNdMin
Dadam.cc150 void Adam::set_use_locking(const bool use_locking) { (void)this->AddAttr(kUseLocking, api::MakeValu… in set_use_locking() function in mindspore::ops::Adam
Dsparse_apply_proximal_gradient_descent.cc108 void SparseApplyProximalGradientDescent::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::SparseApplyProximalGradientDescent
Dsparse_apply_centered_rms_prop.cc145 void SparseApplyCenteredRMSProp::set_use_locking(bool use_locking) { in set_use_locking() function in mindspore::ops::SparseApplyCenteredRMSProp
Dsparse_apply_adagrad.cc141 void SparseApplyAdagrad::set_use_locking(bool use_locking) { in set_use_locking() function in mindspore::ops::SparseApplyAdagrad
Dsparse_apply_momentum.cc118 void SparseApplyMomentum::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::SparseApplyMomentum
Dapply_momentum.cc57 void ApplyMomentum::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::ApplyMomentum
Dsparse_apply_proximal_adagrad.cc137 void SparseApplyProximalAdagrad::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::SparseApplyProximalAdagrad
Dsparse_apply_adagrad_v2.cc145 void SparseApplyAdagradV2::set_use_locking(bool use_locking) { in set_use_locking() function in mindspore::ops::SparseApplyAdagradV2
Dapply_adam_with_amsgrad.cc130 void ApplyAdamWithAmsgrad::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::ApplyAdamWithAmsgrad
Dapply_adam_with_amsgradv2.cc134 void ApplyAdamWithAmsgradV2::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::ApplyAdamWithAmsgradV2
Dfused_sparse_proximal_adagrad.cc132 void FusedSparseProximalAdagrad::set_use_locking(bool use_locking) { in set_use_locking() function in mindspore::ops::FusedSparseProximalAdagrad
Dsparse_apply_r_m_s_prop.cc169 void SparseApplyRMSProp::set_use_locking(const bool use_locking) { in set_use_locking() function in mindspore::ops::SparseApplyRMSProp
Dfused_sparse_adam.cc131 void FusedSparseAdam::set_use_locking(bool use_locking) { in set_use_locking() function in mindspore::ops::FusedSparseAdam
Dfused_sparse_lazy_adam.cc128 void FusedSparseLazyAdam::set_use_locking(bool use_locking) { in set_use_locking() function in mindspore::ops::FusedSparseLazyAdam
Dsparse_apply_ftrl.cc126 void SparseApplyFtrl::set_use_locking(bool use_locking) { in set_use_locking() function in mindspore::ops::SparseApplyFtrl
Dfused_sparse_ftrl.cc156 void FusedSparseFtrl::set_use_locking(bool use_locking) { in set_use_locking() function in mindspore::ops::FusedSparseFtrl
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.cc41 void AdamWeightDecay::set_use_locking(const bool &use_locking) { (void)this->AddAttr("u… in set_use_locking() function in mindspore::ops::AdamWeightDecay