Home
last modified time | relevance | path

Searched refs:mul_op (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/ops/_grad_experimental/
Dgrad_math_ops.py61 mul_op = P.Mul()
67 dstart = mul_op(dout, 1 - weight)
68 dend = mul_op(dout, weight)
69 dweight = mul_op(dout, sub_op(end, start))
Dgrad_array_ops.py29 mul_op = P.Mul()
35 dinput = mul_op(dout, (1 - mask))
36 dvalue = mul_op(dout, mask)
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dscale_npu.cc128 auto mul_op = reinterpret_cast<hiai::op::Mul *>(cur_op); in ConvertScaleToMul() local
129 mul_op->set_input_x1(*npu_inputs.at(INPUT_INDEX)); in ConvertScaleToMul()
131 mul_op->set_input_x2(*npu_inputs.at(SCALE_INDEX)); in ConvertScaleToMul()
158 mul_op->set_input_x2(*reshape_); in ConvertScaleToMul()
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/
Delementwise_tensorrt.cc194 auto mul_op = op_primitive_->value_as_MulFusion(); in AddActivation() local
195 if (mul_op == nullptr) { in AddActivation()
199 activation = mul_op->activation_type(); in AddActivation()
/third_party/mindspore/mindspore/explainer/
D_operators.py175 mul_op = op.Mul()
176 outputs = mul_op(inputs_x, inputs_y)
/third_party/mesa3d/src/gallium/drivers/lima/ir/gp/
Dcodegen.h156 gpir_codegen_mul_op mul_op : 3; member
Ddisasm.c252 switch (instr->mul_op) { in print_mul()
267 if (instr->mul_op == gpir_codegen_mul_op_complex2) in print_mul()
350 fprintf(fp, "\tunknown%u.m01 ", instr->mul_op); in print_mul()
Dcodegen.c126 code->mul_op = gpir_codegen_mul_op_complex1; in gpir_codegen_mul0_slot()
132 code->mul_op = gpir_codegen_mul_op_complex2; in gpir_codegen_mul0_slot()
138 code->mul_op = gpir_codegen_mul_op_select; in gpir_codegen_mul0_slot()
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_math_ops.py235 mul_op = P.Mul()
239 bc_y = neg(mul_op(bc_x, out))
250 mul_op = P.Mul()
254 bc_y = neg(mul_op(bc_x, out))
265 mul_op = P.Mul()
269 bc_y = neg(mul_op(bc_x, out))
/third_party/mesa3d/src/compiler/nir/
Dnir_loop_analyze.c810 nir_op mul_op; in test_iterations() local
815 mul_op = nir_op_fmul; in test_iterations()
821 mul_op = nir_op_imul; in test_iterations()
832 eval_const_binop(mul_op, bit_size, iter_src, step, execution_mode); in test_iterations()
/third_party/mesa3d/src/gallium/drivers/vc4/
Dvc4_qpu_schedule.c327 uint32_t mul_op = QPU_GET_FIELD(inst, QPU_OP_MUL); in calculate_deps() local
349 if (mul_op != QPU_M_NOP) { in calculate_deps()