Searched refs:CalFusedAdam (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_fused_adam.py | 119 def CalFusedAdam(beta1, beta2, one_sub_beta_1, one_sub_beta_2, gradient, eps, weight_decay_tensor, … function
148 param_expect, m_expect, v_expect = CalFusedAdam(
177 param_expect, m_expect, v_expect = CalFusedAdam(
207 param_expect, m_expect, v_expect = CalFusedAdam(
|