| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | matmul_add_fusion.cc | 33 bool IsPrimitiveProper(const CNodePtr &add_cnode, const CNodePtr &matmul_cnode, int index) { in IsPrimitiveProper()
83 int CalNewCnodeBias(const AnfNodePtr &add_weight_node, const CNodePtr &matmul_cnode) { in CalNewCnodeBias()
139 auto matmul_cnode = add_cnode->input(index)->cast<CNodePtr>(); in Run() local
|
| 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()
209 auto matmul_cnode = matmul_node->cast<CNodePtr>(); in Process() local
|
| D | matmul_activation_fusion.cc | 58 auto matmul_cnode = act_cnode->input(1)->cast<CNodePtr>(); in Process() local
|
| D | matmul_allreduce_fusion.cc | 156 auto matmul_cnode = allreduce_cnode->input(kInputIndexOne)->cast<CNodePtr>(); in CreateMatMulAllReduceNode() local
200 auto matmul_cnode = biasAdd_cnode->input(kInputIndexOne)->cast<CNodePtr>(); in CreateMatMulBiasAddAllReduceNode() local
247 auto matmul_cnode = dequant_cnode->input(kInputIndexOne)->cast<CNodePtr>(); in CreateMatMulDequantAllReduceNode() local
|
| D | tensor_dot_fusion.cc | 139 auto matmul_cnode = reshape_1_cnode->input(1)->cast<CNodePtr>(); in Process() local
|
| D | adjust_matmul_pass.cc | 256 auto matmul_cnode = func_graph->NewCNode(dst_prim, {squeeze_cnode, matmul_weight_input}); in AdjustBMMToMM() local
|
| D | batchmatmul_fusion.cc | 381 auto matmul_cnode = func_graph->NewCNode(matmul_inputs); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/ |
| D | overlap_gradmatmul_and_gradallreduce.cc | 45 auto matmul_cnode = node->cast<CNodePtr>(); in ExtractForwardMatMul() local
182 auto matmul_cnode = matmul_node->cast<CNodePtr>(); in ExtractForwardNodes() local
204 auto matmul_cnode = node->cast<CNodePtr>(); in ExtractBackwardNodes() local
|
| D | pass_utils.cc | 102 auto matmul_cnode = node->cast<CNodePtr>(); in ExtractBackwardMatMul() local
|
| D | split_matmul_comm_elementwise_fp.cc | 89 auto matmul_cnode = comm_cnode->input(kIndex1)->cast<CNodePtr>(); in SplitIntoInterleaved() local
|
| D | split_layernorm_comm_fp.cc | 39 bool IsAnyMatMulInputTranspose(const CNodePtr &matmul_cnode) { in IsAnyMatMulInputTranspose()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | inference_matmul_split_fusion.cc | 205 auto matmul_cnode = reshape_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateMatmulSplitNode() local
248 auto matmul_cnode = biasAdd_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateMatmulBiasAddSplitNode() local
376 auto matmul_cnode = reshape_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateMatmulSplitSiluNode() local
430 auto matmul_cnode = biasAdd_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateMatmulBiasAddSplitSiluNode() local
|
| D | mc2_fusion.cc | 184 auto matmul_cnode = reduce_scatter_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateFusionCNode() local
257 auto matmul_cnode = node->cast<CNodePtr>(); in CreateFusionCNode() local
|
| D | matmul_elemwise_fusion.cc | 74 auto matmul_cnode = elemwise_node->input(kIndex1)->cast<CNodePtr>(); in Process() local
|
| D | matmul_allreduce_fusion.cc | 81 auto matmul_cnode = allreduce_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateMatMulAllReduceNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_megatron_op_adjust.cc | 70 auto matmul_cnode = func_graph->NewCNode(matmul_prim_c, inputs); in AdjustMegatronLinearAllGather() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | slice_prepose_pass.cc | 627 const CNodePtr &matmul_cnode, in CreateSlice1ForReshapePrepose()
711 … const CNodePtr &matmul_cnode, const std::vector<int64_t> &shape_in, in PreposeWithAbnormalReshape()
957 auto matmul_cnode = matmul_node->cast<CNodePtr>(); in PreposeWithReshape() local
978 const CNodePtr &matmul_cnode) { in PreposeWithMatmul()
|