Searched defs:mul_cnode (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | matmul_mul_fusion.cc | 33 int CalNewCnodeScale(const CNodePtr &mul_cnode, const CNodePtr &matmul_cnode) { in CalNewCnodeScale()
80 int CalNewCnodeBias(const CNodePtr &mul_cnode, const CNodePtr &matmul_cnode) { in CalNewCnodeBias()
110 bool IsPrimitiveProper(const CNodePtr &mul_cnode, const CNodePtr &matmul_cnode) { in IsPrimitiveProper()
203 auto mul_cnode = node->cast<CNodePtr>(); in Process() local
|
| D | mul_activation_fusion.cc | 62 auto mul_cnode = mul_node->cast<CNodePtr>(); in Process() local
|
| D | sigmoid_mul_fusion.cc | 65 auto mul_cnode = node->cast<CNodePtr>(); in Process() local
|
| D | mul_add_fusion.cc | 240 auto mul_cnode = mul_node->cast<CNodePtr>(); in Process() local
|
| D | mul_reduce_fusion.cc | 271 auto mul_cnode = func_graph->NewCNode(mul_prim, {cnode, second_input_node}); in GenerateMul() local
|
| D | flash_attention_fusion_for_custom.cc | 424 auto mul_cnode = add_cnode->input(1)->cast<CNodePtr>(); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | unused_add_node_remove_pass.cc | 66 auto mul_cnode = add_input->cast<CNodePtr>(); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | momentum_lossscale_fusion.cc | 66 auto mul_cnode = mul->cast<CNodePtr>(); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/ |
| D | smooth_quant.cc | 131 …auto mul_cnode = insert_manager.NewMulNode(func_graph, cnode->input(kInputIndex + kPrimOffset), sc… in InsertScaleForActivation() local
167 …auto mul_cnode = insert_manager.NewAddNode(func_graph, cnode->input(kInputIndex + kPrimOffset), sh… in InsertShiftForActivation() local
|
| D | insert_quant_node_manager.cc | 903 auto mul_cnode = NewMulNode(func_graph, add_cnode, scales_node); in InsertAscendAntiQuantNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_megatron_op_adjust.cc | 126 auto mul_cnode = func_graph->NewCNode(mul_prim_c, {cnode->input(1), scale_param}); in AdjustMegatronScaledMaskedSoftmax() local
|