Home
last modified time | relevance | path

Searched defs:reshape_cnode (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dexpanddims_reshape_fusion.cc38 auto reshape_cnode = node->cast<CNodePtr>(); in CheckCanFuse() local
66 auto reshape_cnode = node->cast<CNodePtr>(); in Process() local
Dreshape_shape_fusion.cc36 auto reshape_cnode = node->cast<CNodePtr>(); in Run() local
Dreshape_transpose_fusion.cc115 std::vector<int> GetShapeOfReshape(const CNodePtr &reshape_cnode, bool *changed = nullptr) { in GetShapeOfReshape()
146 auto reshape_cnode = reshape->cast<CNodePtr>(); in ReshapeTransFusion() local
366 auto reshape_cnode = reshape->cast<CNodePtr>(); in TransReshapeTransFusion() local
Dreshape_reshape_fusion.cc100 AnfNodePtr GenerateNewShapeParam(const CNodePtr &cnode, const CNodePtr &reshape_cnode, const FuncGr… in GenerateNewShapeParam()
Dbatchmatmul_fusion.cc219 auto reshape_cnode = reshape_node->cast<CNodePtr>(); in ResetReshapeParameters() local
Dflash_attention_fusion.cc137 auto reshape_cnode = reshape_node->cast<CNodePtr>(); in GetReshapeParam() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dinference_matmul_split_fusion.cc88 auto reshape_cnode = reshape_node->cast<CNodePtr>(); in GetSplitFusionPatternName() local
200 auto reshape_cnode = split_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateMatmulSplitNode() local
241 auto reshape_cnode = split_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateMatmulBiasAddSplitNode() local
286 auto reshape_cnode = split_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateQuantbatchmatmulSplitNode() local
372 auto reshape_cnode = split_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateMatmulSplitSiluNode() local
423 auto reshape_cnode = split_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateMatmulBiasAddSplitSiluNode() local
480 auto reshape_cnode = split_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateQuantbatchmatmulSplitSiluNode() local
Dtranspose_reshape_fusion.cc42 auto reshape_cnode = CheckAnfNodeIfCNodeAndInputSize(node, kBackendReshapeInputTensorNum); in Process() local
Dreshape_transpose_fusion.cc97 …auto reshape_cnode = CheckAnfNodeIfCNodeAndInputSize(transpose_cnode->input(1), kBackendReshapeInp… in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dmaketuple_unify_mindir.cc59 auto reshape_cnode = node->cast<CNodePtr>(); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dslice_prepose_pass.cc404 auto reshape_cnode = graph->NewCNode({value_node, preceed_cnode, shape_node}); in CreateReshapeCNode() local
562const CNodePtr &reshape_cnode, const std::vector<int64_t> &shape_in, in PreposeWithNormalReshape()
919 const CNodePtr &reshape_cnode) { in PreposeWithReshape()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dflash_attention_score_info.cc1045 auto reshape_cnode = reshape_node->cast<CNodePtr>(); in ReplaceNodeInputOrAttrs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_utils.cc1115 Status UpdateReshapeShapeValue(const CNodePtr &reshape_cnode, const CNodePtr &shape_cnode, const Sh… in UpdateReshapeShapeValue()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc1229 auto reshape_cnode = func_graph->NewCNode(op_inputs); in GenReshapeNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dparameter_manager.cc1120 void HandleAdaSumReshape(const CNodePtr &reshape_cnode, const std::shared_ptr<TensorLayout> &target… in HandleAdaSumReshape()