Home
last modified time | relevance | path

Searched defs:matmul_node (Results 1 – 21 of 21) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/optimizer/fusion/
Dmatmul_mul_fusion_test.cc43 auto matmul_node = std::make_unique<schema::CNodeT>(); in BuildMatMul() local
72 CNodeTptr matmul_node = BuildMatMul(); in BuildGraph() local
Dtrans_matmul_fusion_test.cc119 auto matmul_node = std::make_unique<schema::CNodeT>(); in BuildGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dpass_utils.cc39 bool IsDxMatMul(const CNodePtr &matmul_node) { in IsDxMatMul()
75 bool IsDwMatMul(const CNodePtr &matmul_node) { in IsDwMatMul()
Dbias_add_comm_swap.cc64 auto matmul_node = GetInputNodeWithFilter(node, [&](const CNodePtr &cnode) { in FindMatMulNode() local
91 auto matmul_node = FindMatMulNode(comm_node); in FindValidCommNode() local
Dassign_add_opt.cc80 bool GetMatMulTransposeValue(const CNodePtr &matmul_node, const std::string &attr_name) { in GetMatMulTransposeValue()
292 auto matmul_node = assign_add_node->cast<CNodePtr>()->input(kIndex2)->cast<CNodePtr>(); in SkipAssignAddEliminate() local
297 auto matmul_node = cast_node->input(1)->cast<CNodePtr>(); in SkipAssignAddEliminate() local
Doverlap_gradmatmul_and_gradallreduce.cc178 auto matmul_node = RealInputNode(node, 1); in ExtractForwardNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/optimizer/fusion/
Dmatmul_activation_fusion_pass.cc62 auto &matmul_node = graph->nodes.at(matmul_index); in DoFusion() local
Dmatmul_add_fusion_pass.cc93 auto &matmul_node = graph->nodes.at(matmul_index); in DoFusion() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/
Dmatmul_biasadd_fusion.cc88 auto matmul_node = matmul->cast<CNodePtr>(); in Process() local
Dmatmul_biasadd_relu_fusion.cc91 auto matmul_node = matmul->cast<CNodePtr>(); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Daffine_fusion.cc59 auto matmul_node = node->cast<CNodePtr>(); in Process() local
Dmatmul_mul_fusion.cc208 auto matmul_node = mul_cnode->input(1); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/optimizer/fusion/fusion_inout_test/
Dmatmul_act_fusion_inout_test.cc36 auto matmul_node = AddMatmul(graph_, "matmul_node"); in InitGraph() local
Dmatmul_mul_fusion_inout_test.cc36 auto matmul_node = AddMatmul(graph_, "matmul_node"); in InitGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dcast_matmul_fusion.cc28 void UpdateBuildInfo(const AnfNodePtr &matmul_node, const AnfNodePtr &cast_node) { in UpdateBuildInfo()
Dinsert_pad.cc190 void UpdateMatmulInfo(const AnfNodePtr &matmul_node, const vec &unpad_shape, const vec &tail_shape)… in UpdateMatmulInfo()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_einsum_adjust.cc142 ops::MatMulFusion matmul_node; in Adjust() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/
Dinsert_depend_for_grad_comm.cc135 bool IsDxMatMul(const CNodePtr &matmul_node) { in IsDxMatMul()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Ddynamic_gru_v2_grad_fission.cc180 auto matmul_node = NewCNode(matmul_inputs, func_graph); in AddTLoopNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dslice_prepose_pass.cc951 auto matmul_node = reshape_cnode->input(1); in PreposeWithReshape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc3228 auto matmul_node = dyn_cast<CNode>(node->input(kOneNum)); in ExportPrimDense() local
4250 auto matmul_node = dyn_cast<CNode>(node->input(kOneNum)); in ExportMergeGemm() local