Home
last modified time | relevance | path

Searched defs:mul_node (Results 1 – 23 of 23) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/optimizer/fusion/
Dmatmul_mul_fusion_test.cc56 auto mul_node = std::make_unique<schema::CNodeT>(); in BuildMul() local
74 auto mul_node = BuildMul(); in BuildGraph() local
Dconstant_folding_fusion_test.cc157 auto mul_node = std::make_unique<schema::CNodeT>(); in BuildMixGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dmul_activation_fusion.cc60 auto mul_node = act_cnode->input(1); in Process() local
Dprelu_fusion.cc88 auto mul_node = node->cast<CNodePtr>()->input(kInputIndexOne); in Process() local
Dgegluv2_fusion.cc125 auto mul_node = node->cast<CNodePtr>(); in Process() local
Dmul_add_fusion.cc237 …auto mul_node = utils::isa<CNodePtr>(add_cnode->input(SECOND_INPUT)) ? add_cnode->input(SECOND_INP… in Process() local
Dadjust_resize_dims_pass.cc438 auto mul_node = CreateMulNode(func_graph, input_shape_cnode); in AdjustResize5DToResize4D() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc47 bool CheckMulInputShapeEqual(const CNodePtr &mul_node) { in CheckMulInputShapeEqual()
324 …reateTile(const FuncGraphPtr &graph, const CNodePtr &sparse_softmax_node, const CNodePtr &mul_node, in CreateTile()
424 CNodePtr GetDependNode(const CNodePtr &mul_node) { in GetDependNode()
467 auto mul_node = pass.NewCNode(mul_input, graph); in CreateMul() local
504 CNodePtr CreateMulInput(const FuncGraphPtr &graph, const CNodePtr &mul_node, const AnfNodePtr &spar… in CreateMulInput()
594 auto mul_node = node->cast<CNodePtr>(); in Process() local
686 auto mul_node = CreateMul(graph, sparse_softmax_node_grad, softmax_node_outputs[1], *this); in Process() local
752 auto mul_node = node->cast<CNodePtr>(); in Process() local
809 auto mul_node = node->cast<CNodePtr>(); in Process() local
Drenorm_split.cc102 auto mul_node = func_graph->NewCNode(mul_inputs); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/legacy_optimizer/fusion/
Dmatmul_biasadd_fusion_pass.cc70 auto &mul_node = graph->nodes.at(mul_index); in DoFusion() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Drandom_normal_mapper.cc106 … auto mul_node = NewCNode(cnode, prim::kPrimMul, {cnode, scale_param}, cnode->abstract()->Clone(), in Mapper() local
Dresize_mapper.cc205 …auto mul_node = NewCNode(cnode, prim::kPrimMul, {cast_fp32_node, scale_input}, {DIMENSION_2D}, kNu… in CalResizeShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/optimizer/fusion/fusion_inout_test/
Dmatmul_mul_fusion_inout_test.cc41 auto mul_node = AddMul(graph_, matmul_node, "mul_node"); in InitGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dconv2d_backprop_filter_mul_fission.cc121 CNodePtr mul_node = NewCNode(mul_inputs, func_graph); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dbatchnorm_transform.cc113 …auto mul_node = fg->NewCNode({NewValueNode(std::make_shared<Primitive>(kMulOpName)), data_node, su… in CreateMulNode() local
122 auto mul_node = inputs[1]; in CreateAssignSubNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/
Dir_pass.cc168 CNodePtr Run(const CNodePtr &mul_node, const AnfNodePtr &sparse_softmax_node) { in Run()
299 CNodePtr CreateTile(const CNodePtr &sparse_softmax_node, const CNodePtr &mul_node) { in CreateTile()
392 CNodePtr CreateMulInput(const CNodePtr &mul_node, const AnfNodePtr &sparse_softmax_node, in CreateMulInput()
434 auto mul_node = node->cast<CNodePtr>(); in GradSparseSoftmaxCrossEntropyWithLogitsUnifyMindIR() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dkvcache_quant_pass.cc196 auto mul_node = add_node->cast<CNodePtr>()->input(kIndexOne); in RunQuantPass() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Darithmetic_simplify.cc87 auto mul_node = node->cast<CNodePtr>()->inputs()[0]; in operator ()() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dbias_add_comm_swap.cc204 auto mul_node = fg->NewCNode(mul_node_inputs); in HandleNodeBiasAdd() local
Dmatmul_add_comm_reduction.cc234 auto mul_node = fg->NewCNode(mul_node_inputs); in HandleNodeBiasAdd() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Dflash_sp.cc244 auto mul_node = left_node->func_graph()->NewCNode(mul_inputs); in NewMulNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dquantize_util.cc1202 auto mul_node = cnode->input(index); in RemoveAntiQuantModeNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc3696 auto mul_node = "Atan2_" + node_name + "_mul"; in ExportPrimAtan2() local