| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_adam_fusion.py | 35 self.op_mul = P.Mul()
52 next_m = self.op_mul(beta1, m_fp32) + \
53 … self.op_mul(self.op_cast(F.tuple_to_array((1.0,)), mstype.float32) - beta1, gradient_fp32)
54 …next_v = self.op_mul(beta2, v_fp32) + self.op_mul(self.op_cast(F.tuple_to_array((1.0,)), mstype.fl…
58 update = self.op_mul(weight_decay_tensor, param_fp32) + update
59 update_with_lr = self.op_mul(lr, update)
|
| /third_party/mindspore/tests/st/fl/mobile/src/ |
| D | adam.py | 70 op_mul = P.Mul()
84 next_m = op_mul(beta1, m_fp32) + op_select(cond, m_fp32,\
85 op_mul(op_cast(F.tuple_to_array((1.0,)), mstype.float32) - beta1, gradient_fp32))
87 next_v = op_mul(beta2, v_fp32) + op_select(cond, v_fp32,\
88 … op_mul(op_cast(F.tuple_to_array((1.0,)), mstype.float32) - beta2, op_square(gradient_fp32)))
92 update = op_mul(weight_decay, param_fp32) + update
94 update_with_lr = op_mul(lr, update)
127 op_mul = P.Mul()
132 F.assign(m, op_mul(beta1, m))
133 F.assign(v, op_mul(beta2, v))
[all …]
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | adam.py | 71 op_mul = P.Mul()
85 next_m = op_mul(beta1, m_fp32) + op_select(cond, m_fp32,\
86 op_mul(op_cast(F.tuple_to_array((1.0,)), mstype.float32) - beta1, gradient_fp32))
88 next_v = op_mul(beta2, v_fp32) + op_select(cond, v_fp32,\
89 … op_mul(op_cast(F.tuple_to_array((1.0,)), mstype.float32) - beta2, op_square(gradient_fp32)))
93 update = op_mul(weight_decay, param_fp32) + update
95 update_with_lr = op_mul(lr, update)
128 op_mul = P.Mul()
133 F.assign(m, op_mul(beta1, m))
134 F.assign(v, op_mul(beta2, v))
[all …]
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | adam.py | 70 op_mul = P.Mul()
84 next_m = op_mul(beta1, m_fp32) + op_select(cond, m_fp32,\
85 op_mul(op_cast(F.tuple_to_array((1.0,)), mstype.float32) - beta1, gradient_fp32))
87 next_v = op_mul(beta2, v_fp32) + op_select(cond, v_fp32,\
88 … op_mul(op_cast(F.tuple_to_array((1.0,)), mstype.float32) - beta2, op_square(gradient_fp32)))
92 update = op_mul(weight_decay, param_fp32) + update
94 update_with_lr = op_mul(lr, update)
127 op_mul = P.Mul()
132 F.assign(m, op_mul(beta1, m))
133 F.assign(v, op_mul(beta2, v))
[all …]
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | adam.py | 59 op_mul = P.Mul()
70 next_m = op_mul(beta1, m_fp32) + op_mul(op_cast(F.tuple_to_array((1.0,)), mstype.float32)
73 next_v = op_mul(beta2, v_fp32) + op_mul(op_cast(F.tuple_to_array((1.0,)), mstype.float32)
78 update = op_mul(weight_decay, param_fp32) + update
80 update_with_lr = op_mul(lr, update)
112 op_mul = P.Mul()
117 success = F.depend(success, F.assign(m, op_mul(beta1, m)))
118 success = F.depend(success, F.assign(v, op_mul(beta2, v)))
125 op_mul(F.tuple_to_array((1.0,)) - beta1, grad_value))
129 op_mul(F.tuple_to_array((1.0,)) - beta2, op_square(grad_value)))
[all …]
|
| D | lamb.py | 61 op_mul = P.Mul()
80 … next_m = op_mul(beta1, m_fp32) + op_mul(op_cast(num_one, mstype.float32) - beta1, gradient_fp32)
82 …next_v = op_mul(beta2, v_fp32) + op_mul(op_cast(num_one, mstype.float32) - beta2, op_square(gradie…
91 g_norm_hat = op_norm(op_mul(next_mm, op_rsqrt(next_vv + eps)) + weight_decay * param_fp32)
103 update = update + op_mul(weight_decay, param_fp32)
105 update_with_lr = op_mul(op_mul(trust_ratio, lr), update)
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_fused_adam.py | 33 self.op_mul = P.Mul()
53 next_m = self.op_mul(beta1, m_fp32) + \
54 self.op_mul(self.op_cast(one_sub_beta_1,
56 next_v = self.op_mul(beta2, v_fp32) + self.op_mul(self.op_cast(one_sub_beta_2,
60 update = self.op_mul(weight_decay_tensor, param_fp32) + update
61 update_with_lr = self.op_mul(lr, update)
75 self.op_mul = P.Mul()
99 next_m = self.op_mul(beta1, m_fp32) + \
100 self.op_mul(self.op_cast(one_sub_beta_1,
102 next_v = self.op_mul(beta2, v_fp32) + self.op_mul(self.op_cast(one_sub_beta_2,
[all …]
|
| /third_party/boost/libs/spirit/example/karma/ |
| D | calc2_ast_vm.hpp | 115 op_mul, // multiply top two stack entries enumerator
161 expr = binary_op(op_mul, expr, rhs); in operator *=()
|
| /third_party/boost/libs/spirit/example/x3/calc/ |
| D | calc6.cpp | 105 op_mul, // multiply top two stack entries enumerator
152 case op_mul: in execute()
194 case '*': code.push_back(op_mul); break; in operator ()()
|
| /third_party/mesa3d/src/gallium/drivers/vc4/ |
| D | vc4_qpu_disasm.c | 398 uint32_t op_mul = QPU_GET_FIELD(inst, QPU_OP_MUL); in print_mul_op() local
400 bool is_mov = (op_mul == QPU_M_V8MIN && in print_mul_op()
407 fprintf(stderr, "%s", DESC(qpu_mul_opcodes, op_mul)); in print_mul_op()
412 if (op_mul != QPU_M_NOP) in print_mul_op()
|
| /third_party/mindspore/tests/ut/python/ir/ |
| D | test_row_tensor.py | 169 op_mul = P.Mul()
181 …next_m = op_mul(beta1, m_fp32) + op_mul(op_cast(F.tuple_to_array((1.0,)), mstype.float32) - beta1,…
183 next_v = op_mul(beta2, v_fp32) + op_mul(op_cast(F.tuple_to_array((1.0,)), mstype.float32)
188 update = update + op_mul(weight_decay_tensor, param_fp32)
190 update_with_lr = op_mul(lr, update)
|
| /third_party/boost/libs/spirit/example/qi/compiler_tutorial/ |
| D | calc6.cpp | 120 op_mul, // multiply top two stack entries enumerator
167 case op_mul: in execute()
209 case '*': code.push_back(op_mul); break; in operator ()()
|
| /third_party/boost/libs/spirit/example/x3/calc/calc7/ |
| D | vm.hpp | 22 op_mul, // multiply top two stack entries enumerator
|
| D | vm.cpp | 34 case op_mul: in execute()
|
| /third_party/boost/libs/spirit/example/x3/calc/calc8/ |
| D | compiler.cpp | 84 case op_mul: in print_assembler()
137 case '*': program.op(op_mul); break; in operator ()()
|
| D | vm.hpp | 22 op_mul, // multiply top two stack entries enumerator
|
| D | vm.cpp | 35 case op_mul: in execute()
|
| /third_party/boost/libs/spirit/example/qi/compiler_tutorial/calc7/ |
| D | vm.hpp | 22 op_mul, // multiply top two stack entries enumerator
|
| D | compiler.cpp | 85 case op_mul: in print_assembler()
139 case '*': program.op(op_mul); break; in operator ()()
|
| D | vm.cpp | 35 case op_mul: in execute()
|
| /third_party/boost/boost/python/detail/ |
| D | operator_id.hpp | 14 op_mul, enumerator
|
| /third_party/boost/libs/spirit/example/qi/compiler_tutorial/calc8/ |
| D | vm.hpp | 22 op_mul, // multiply top two stack entries enumerator
|
| /third_party/boost/libs/spirit/example/x3/calc/calc9/ |
| D | vm.hpp | 22 op_mul, // multiply top two stack entries enumerator
|
| /third_party/boost/libs/spirit/example/qi/compiler_tutorial/conjure1/ |
| D | vm.hpp | 22 op_mul, // multiply top two stack entries enumerator
|
| /third_party/boost/libs/spirit/example/qi/compiler_tutorial/mini_c/ |
| D | vm.hpp | 22 op_mul, // multiply top two stack entries enumerator
|