Home
last modified time | relevance | path

Searched defs:output_cnode (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Drecompute_prepare.h62 auto output_cnode = output->cast<CNodePtr>(); in GetRealOutputNodes() local
Dincorporate_getitem.h157 const auto &output_cnode = output->cast<CNodePtr>(); in GetOutputSize() local
309 const auto &output_cnode = output->cast<CNodePtr>(); in AlwaysTransformThisIndex() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dlayer_norm_beta_gamma_backprop_fusion.cc35 auto output_cnode = output->cast<CNodePtr>(); in GetOutputCastNodes() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Drectify_do_mask_kernel_info.cc53 auto output_cnode = output_node.first->cast<CNodePtr>(); in Process() local
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dmindspore_importer.cc136 auto output_cnode = utils::cast<CNodePtr>(output_node); in GetFuncGraphOutputName() local
Dmindir_control_flow_adjust.cc51 auto output_cnode = output_node->cast<CNodePtr>(); in GetFgOutput() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dadd_tensor_array.cc67 auto output_cnode = output_node->cast<CNodePtr>(); in SetGraphOutput() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Drecompute.cc151 auto output_cnode = output_node->cast<CNodePtr>(); in GetMaxSubGraph() local
319 auto output_cnode = output_node->cast<CNodePtr>(); in SetRecomputedAttr() local
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dcomposite.cc777 CNodePtr output_cnode = fg_ptr->NewCNodeInOrder({NewValueNode(fgcond_ptr), fn, resl}); in GenerateFuncGraph() local
819 …AnfNodePtr output_cnode = fg_ptr->NewCNodeInOrder({NewValueNode(prim::kPrimSwitch), NewValueNode(s… in MakeCond() local
852 CNodePtr output_cnode = fg_ptr->NewCNodeInOrder({NewValueNode(fgcond_ptr), fn, resl}); in MakeNext() local
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Danf_exporter.cc599 …fExporter::ConvertInputCNodeCommonOp(const AnfNodePtr &input_anode, schema::CNodeT *output_cnode) { in ConvertInputCNodeCommonOp()
627 …ter::ConvertInputCNode(const std::shared_ptr<AnfNode> &input_anode, schema::CNodeT *output_cnode) { in ConvertInputCNode()
/third_party/mindspore/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc283 auto output_cnode = utils::cast<CNodePtr>(output_node); in InitSubGraphOutNode() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_session.cc293 auto output_cnode = user.first->cast<CNodePtr>(); in UpdatedByAssign() local
Dkernel_graph.cc924 auto output_cnode = output_node.first->cast<CNodePtr>(); in ReplaceNode() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dub_pattern_fusion.cc374 auto output_cnode = output->cast<CNodePtr>(); in SetFusionOpRefInfos() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dkprim.cc362 auto output_cnode = output->cast<CNodePtr>(); in AppendMonadOutput() local
Ddfunctor.cc434 auto output_cnode = output_node->cast<CNodePtr>(); in MapMorphism() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc525 auto output_cnode = output->cast<CNodePtr>(); in FindPipelineCareNode() local