| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | scatter_nd_arithmetic.cc | 101 void ScatterNdUpdate::Init(const bool use_locking) { this->set_use_locking(use_locking); } in Init()
103 void ScatterNdUpdate::set_use_locking(const bool use_locking) { in set_use_locking()
112 void ScatterNdAdd::Init(const bool use_locking) { this->set_use_locking(use_locking); } in Init()
114 void ScatterNdAdd::set_use_locking(const bool use_locking) { in set_use_locking()
123 void ScatterNdSub::Init(const bool use_locking) { this->set_use_locking(use_locking); } in Init()
125 void ScatterNdSub::set_use_locking(const bool use_locking) { in set_use_locking()
134 void ScatterNdMul::Init(const bool use_locking) { this->set_use_locking(use_locking); } in Init()
136 void ScatterNdMul::set_use_locking(const bool use_locking) { in set_use_locking()
145 void ScatterNdDiv::Init(const bool use_locking) { this->set_use_locking(use_locking); } in Init()
147 void ScatterNdDiv::set_use_locking(const bool use_locking) { in set_use_locking()
[all …]
|
| D | adam.cc | 145 void Adam::Init(const bool use_locking, const bool use_nesterov) { in Init()
150 void Adam::set_use_locking(const bool use_locking) { (void)this->AddAttr(kUseLocking, api::MakeValu… in set_use_locking()
|
| D | sparse_apply_proximal_gradient_descent.cc | 106 void SparseApplyProximalGradientDescent::Init(const bool use_locking) { this->set_use_locking(use_l… in Init()
108 void SparseApplyProximalGradientDescent::set_use_locking(const bool use_locking) { in set_use_locking()
|
| D | sparse_apply_centered_rms_prop.cc | 143 void SparseApplyCenteredRMSProp::Init(bool use_locking) { set_use_locking(use_locking); } in Init()
145 void SparseApplyCenteredRMSProp::set_use_locking(bool use_locking) { in set_use_locking()
|
| D | sparse_apply_adagrad.cc | 125 void SparseApplyAdagrad::Init(float lr, bool update_slots, bool use_locking) { in Init()
141 void SparseApplyAdagrad::set_use_locking(bool use_locking) { in set_use_locking()
|
| D | sparse_apply_momentum.cc | 113 void SparseApplyMomentum::Init(const bool use_locking, const bool use_nesterov) { in Init()
118 void SparseApplyMomentum::set_use_locking(const bool use_locking) { in set_use_locking()
|
| D | apply_momentum.cc | 47 void ApplyMomentum::Init(const bool use_nesterov, const bool use_locking, const float gradient_scal… in Init()
57 void ApplyMomentum::set_use_locking(const bool use_locking) { in set_use_locking()
|
| D | sparse_apply_proximal_adagrad.cc | 135 void SparseApplyProximalAdagrad::Init(const bool use_locking) { this->set_use_locking(use_locking);… in Init()
137 void SparseApplyProximalAdagrad::set_use_locking(const bool use_locking) { in set_use_locking()
|
| D | sparse_apply_adagrad_v2.cc | 124 void SparseApplyAdagradV2::Init(float lr, float epsilon, bool update_slots, bool use_locking) { in Init()
145 void SparseApplyAdagradV2::set_use_locking(bool use_locking) { in set_use_locking()
|
| D | fused_sparse_proximal_adagrad.cc | 132 void FusedSparseProximalAdagrad::set_use_locking(bool use_locking) { in set_use_locking()
141 void FusedSparseProximalAdagrad::Init(bool use_locking) { this->set_use_locking(use_locking); } in Init()
|
| D | fused_sparse_adam.cc | 131 void FusedSparseAdam::set_use_locking(bool use_locking) { in set_use_locking()
149 void FusedSparseAdam::Init(bool use_locking, bool use_nesterov) { in Init()
|
| D | fused_sparse_lazy_adam.cc | 128 void FusedSparseLazyAdam::set_use_locking(bool use_locking) { in set_use_locking()
146 void FusedSparseLazyAdam::Init(bool use_locking, bool use_nesterov) { in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/optim/ |
| D | adam.py | 44 def _run_lazy_opt_with_sparse_dist(opt, sparse_opt, use_locking, use_nesterov, target, beta1_power, argument
89 def _run_map_tensor_lazy_opt_with_sparse_dist(opt, sparse_opt, use_locking, use_nesterov, target, argument
129 def _run_lazy_opt_with_one_number_dist(opt, sparse_opt, use_locking, use_nesterov, target, argument
146 def _run_lazy_opt_with_sparse(opt, sparse_opt, use_locking, use_nesterov, target, beta1_power, beta… argument
184 def _run_map_tensor_lazy_opt_with_sparse(opt, sparse_opt, use_locking, use_nesterov, target, beta1_… argument
216 def _run_lazy_opt_with_one_number(opt, sparse_opt, use_locking, use_nesterov, target, beta1_power, argument
284 def _run_opt_with_sparse_dist(opt, sparse_opt, use_locking, use_nesterov, target, beta1_power, argument
343 def _run_opt_with_one_number_dist(opt, sparse_opt, use_locking, use_nesterov, target, argument
360 def _run_opt_with_sparse(opt, sparse_opt, use_locking, use_nesterov, target, argument
414 def _run_opt_with_one_number(opt, sparse_opt, use_locking, use_nesterov, target, argument
[all …]
|
| D | lazyadam.py | 38 def _run_opt_with_sparse_dist(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, beta1… argument
91 def _run_map_tensor_opt_with_sparse_dist(opt, sparse_opt, push, pull, use_locking, use_nesterov, ta… argument
131 def _run_opt_with_one_number_dist(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, argument
153 def _run_opt_with_sparse(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, beta1_powe… argument
201 def _run_map_tensor_opt_with_sparse(opt, sparse_opt, push, pull, use_locking, use_nesterov, target,… argument
234 def _run_opt_with_one_number(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, beta1_… argument
417 …def __init__(self, params, learning_rate=1e-3, beta1=0.9, beta2=0.999, eps=1e-8, use_locking=False, argument
502 def _init_distributed_opts(self, use_locking, use_nesterov): argument
|
| D | proximal_ada_grad.py | 45 def _check_param_value(accum, l1, l2, use_locking, prim_name=None): argument
192 use_locking=False, loss_scale=1.0, weight_decay=0.0): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_scatter_op.py | 29 def __init__(self, input_shape, input_dtype, use_locking): argument
40 def __init__(self, input_shape, input_dtype, use_locking): argument
|
| D | test_sparse_apply_adagrad_da.py | 29 def __init__(self, use_locking=False): argument
|
| D | test_sparse_apply_r_m_s_prop_op.py | 30 def __init__(self, rho, momentum, epsilon, use_locking=False): argument
|
| D | test_sparse_apply_centered_rms_prop_op.py | 27 def __init__(self, use_locking=False): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | sparse_apply_centered_rms_prop_impl.cu | 36 …rseApplyCenteredRMSPropUpdate(const size_t size, const size_t indices_size, const bool use_locking, in SparseApplyCenteredRMSPropUpdate()
58 …rseApplyCenteredRMSPropUpdate(const size_t size, const size_t indices_size, const bool use_locking, in SparseApplyCenteredRMSPropUpdate()
81 …rseApplyCenteredRMSPropUpdate(const size_t size, const size_t indices_size, const bool use_locking, in SparseApplyCenteredRMSPropUpdate()
111 …CalSparseApplyCenteredRMSProp(const size_t size, const size_t indices_size, const bool use_locking, in CalSparseApplyCenteredRMSProp()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | nn_ops.py | 2345 def __init__(self, use_nesterov=False, use_locking=False, gradient_scale=1.0): argument
2848 def __init__(self, use_locking=False): argument
2946 def __init__(self, use_locking=False): argument
3708 def __init__(self, use_locking=False, use_nesterov=False): argument
3805 def __init__(self, use_locking=False, use_nesterov=False): argument
3927 def __init__(self, use_locking=False, use_nesterov=False): argument
4055 def __init__(self, use_locking=False, use_nesterov=False): argument
4141 def __init__(self, lr, l1, l2, lr_power, use_locking=False): argument
4240 def __init__(self, use_locking=False): argument
4751 def __init__(self, lr, update_slots=True, use_locking=False): argument
[all …]
|
| D | array_ops.py | 65 def __init__(self, use_locking=False): argument
145 def __init__(self, use_locking=False): argument
2322 def __init__(self, use_locking=True): argument
2392 def __init__(self, use_locking=True): argument
2637 def __init__(self, use_locking=False): argument
2756 def __init__(self, use_locking=False): argument
3053 def __init__(self, use_locking=False): argument
3131 def __init__(self, use_locking=False): argument
3333 def __init__(self, use_locking=False): argument
3402 def __init__(self, use_locking=False): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_ops_applyadamwithamsgradv2.py | 24 def __init__(self, use_locking=False): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_apply_adagrad_dad_op.py | 25 def __init__(self, use_locking=False): argument
|
| D | test_sparse_apply_r_m_s_prop_cpu_op.py | 29 def __init__(self, rho, momentum, epsilon, use_locking=False): argument
|