Home
last modified time | relevance | path

Searched defs:origin_node (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dunsorted_segment_sum_d_fission.cc29 bool CheckInputs(const CNodePtr &origin_node) { in CheckInputs()
50 …r UnsortedSegmentSumDFission::CreatePadding(const FuncGraphPtr &graph, const CNodePtr &origin_node, in CreatePadding()
67 …egmentSumDFission::CreateUnsortedSegmentSum(const FuncGraphPtr &graph, const CNodePtr &origin_node, in CreateUnsortedSegmentSum()
119 auto origin_node = node->cast<CNodePtr>(); in Process() local
Dgather_v2_ds_fission.cc36 bool CheckInputs(const CNodePtr &origin_node) { in CheckInputs()
59 CNodePtr GatherV2DsFission::CreatePad(const FuncGraphPtr &graph, const CNodePtr &origin_node, in CreatePad()
107 CNodePtr GatherV2DsFission::CreateGatherV2Ds(const FuncGraphPtr &graph, const CNodePtr &origin_node, in CreateGatherV2Ds()
164 auto origin_node = node->cast<CNodePtr>(); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dall_to_all_v_for_ge.cc61 MemRange CalcAllToAllvForGEMemRange(const CNodePtr &origin_node, uint32_t rank_size, in CalcAllToAllvForGEMemRange()
106 std::tuple<MemRange, MemRange> CalcAllToAllvForGEInput(const CNodePtr &origin_node, uint32_t rank_s… in CalcAllToAllvForGEInput()
184 …AllvForGENode(const FuncGraphPtr &graph, const AnfNodePtr &input_node, const CNodePtr &origin_node, in CreateAllToAllvForGENode()
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/place/
Dgraph_splitter_test.cc59 auto origin_node = func_graph_->NewCNode(inputs); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Dabstract_actor.cc295 const auto &origin_node = graph->GetRefNodeRecursive(output_node).first; in IsOutputAddressPersisted() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/parse/
Dresolve.cc246 AnfNodePtr ConvertInterpretedObjForResolve(const AnfNodePtr &origin_node, const ValuePtr &convert_r… in ConvertInterpretedObjForResolve()
261 AnfNodePtr ConvertObjectToNode(const AnfNodePtr &origin_node, const py::object &obj, const FuncGrap… in ConvertObjectToNode()
443 bool ResolveObjectToNode(const AnfNodePtr &origin_node, const py::object &obj, AnfNodePtr *const no… in ResolveObjectToNode()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Drecompute_helper.cc489 CNodePtr CreateNewRecomputedNode(const FuncGraphPtr &graph, const CNodePtr &origin_node, in CreateNewRecomputedNode()
524 CNodePtr NewRecomputedNode(const FuncGraphPtr &graph, const CNodePtr &origin_node, in NewRecomputedNode()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dbatchnorm_to_scale_fusion.cc120 int CalEstimatedData(const AnfNodePtr &origin_node, const AnfNodePtr &scale_factor_node) { in CalEstimatedData()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/static_analysis/
Dprogram_specialize.cc60 AnfNodePtr BuildValueNode(const ValuePtr &v, const AnfNodePtr &origin_node, const AbstractBasePtr &… in BuildValueNode()
1116 const auto &origin_node = conf_iter->first->node(); in BuildReplacedNode() local
1896 AnfNodePtr FuncGraphSpecializer::BuildValueNodeForAbstractFunction(const AnfNodePtr &origin_node, in BuildValueNodeForAbstractFunction()
1931 AnfNodePtr FuncGraphSpecializer::BuildPossibleValueNode(const AnfNodePtr &origin_node, const Abstra… in BuildPossibleValueNode()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/graph_compiler/
Ddefault_graph_compiler.cc134 std::vector<AnfNodePtr> DefaultGraphCompiler::SkipMakeTuple(const AnfNodePtr &origin_node) { in SkipMakeTuple()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgenerate_graph.cc217 CNodePtr CreateNewCNodeForReplace(const CNodePtr &origin_node, const PrimitivePtr &new_prim) { in CreateNewCNodeForReplace()
Dgraph_splitter.cc56 ValueNodePtr CreateFakeValueNode(bool use_origin_node, const AnfNodePtr &origin_node, bool use_fake… in CreateFakeValueNode()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dps_embedding_cache_inserter.cc64 ValueNodePtr CreateFakeValueNode(const AnfNodePtr &origin_node) { in CreateFakeValueNode()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dinsert_type_transform_op.cc95 const CNodePtr &origin_node) { in CreateNewNode()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Ddropout_unify_mindir.cc354 void UpdateReturnNode(const FuncGraphPtr &graph, const AnfNodePtr &origin_node, const AnfNodePtr &n… in UpdateReturnNode()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_graph_executor.cc1049 session::AnfWithOutIndex origin_node = std::make_pair(graph_inputs[in_out_index.first], 0); in AddRefCorrespondPairs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dcontrol_node_scheduler.cc70 const auto &origin_node = kernel_graph->GetRefCorrespondOutput({backend_node, index}).first; in is_need_copy_device_tensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/somas/
Dsomas.cc1011 auto origin_node = node_item.second.first; in SummaryInputProcess() local