Searched refs:mul_op (Results 1 – 11 of 11) sorted by relevance
/third_party/mindspore/mindspore/ops/_grad_experimental/ |
D | grad_math_ops.py | 61 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))
|
D | grad_array_ops.py | 29 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/ |
D | scale_npu.cc | 128 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/ |
D | elementwise_tensorrt.cc | 194 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.py | 175 mul_op = op.Mul() 176 outputs = mul_op(inputs_x, inputs_y)
|
/third_party/mesa3d/src/gallium/drivers/lima/ir/gp/ |
D | codegen.h | 156 gpir_codegen_mul_op mul_op : 3; member
|
D | disasm.c | 252 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()
|
D | codegen.c | 126 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/ |
D | grad_math_ops.py | 235 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/ |
D | nir_loop_analyze.c | 810 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/ |
D | vc4_qpu_schedule.c | 327 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()
|