Home
last modified time | relevance | path

Searched defs:backend_node (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/runtime/graph_scheduler/
Dscheduler_helper_test.cc73 auto backend_node = kernel_graph->NewCNode(inputs); in TEST_F() local
Dcontrol_node_parser_test.cc53 auto backend_node = kernel_graph->NewCNode(inputs); in BuildKernelGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dcontrol_node_parser.cc350 …iceTensorForValueNode(const KernelWithIndex &front_node_with_index, const AnfNodePtr &backend_node, in CreateDeviceTensorForValueNode()
703 …ildInfoForFrontNode(const KernelWithIndex &front_node_with_index, const AnfNodePtr &backend_node) { in CreateBuildInfoForFrontNode()
806 KernelWithIndex GetFrontNodeByKernelGraph(const AnfNodePtr &backend_node, const KernelGraph *const … in GetFrontNodeByKernelGraph()
1371 …olNodeParser::IsControlFlowDataArrow(const KernelGraphPtr &graph, const AnfNodePtr &backend_node) { in IsControlFlowDataArrow()
1961 const auto &backend_node = node_with_index_with_context.first.first; in CreateDeviceTensors() local
1986 const auto &backend_node = node_with_index_to_context.first.first; in FetchFrontValueNode() local
Dgraph_compiler.cc509 auto backend_node = kernel_graph->output(); in CompileGraph() local
672 auto backend_node = root_graph->output(); in CompileWholeGraphForGraphRunMode() local
Dcontrol_node_scheduler.cc52 bool is_need_copy_device_tensor(const AnfNodePtr &backend_node, size_t index) { in is_need_copy_device_tensor()
417 const AnfNodePtr &backend_node = node_with_context.second.first.first; in BuildExitActor() local
Dgraph_scheduler.cc3132 const auto &backend_node = node_with_index_with_context.first.first; in PersistDeviceTensorForRootGraphControlNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Ddata_prepare_actor.cc1180 …eActor::CopyDataFromDeviceTensorStore(const AnfNodePtr &front_node, const AnfNodePtr &backend_node, in CopyDataFromDeviceTensorStore()
1232 void DataPrepareActor::PrepareDataForWeightNode(const AnfNodePtr &backend_node, const AnfNodePtr &f… in PrepareDataForWeightNode()
Dany_type_kernel_actor.cc457 size_t FetchInputIndexByBackendParameter(const AnfNodePtr &backend_node, const KernelGraphPtr &fron… in FetchInputIndexByBackendParameter()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dkernel_graph_mgr.cc1137 void AddValueNode(const AnfNodePtr &backend_node, KernelGraph *graph) { in AddValueNode()
1163 const auto &backend_node = graph->GetBackendAnfByFrontAnf(anf); in GetNewCNodeInputs() local
1313 auto backend_node = graph->GetBackendAnfByFrontAnf(node_input); in CreateSwitchInput() local
2658 …lGraphMgr::HandleInternalOutput(const AnfNodePtr &input_front_node, const AnfNodePtr &backend_node, in HandleInternalOutput()
Danf_runtime_algorithm.cc1498 AnfNodePtr AnfRuntimeAlgorithm::FetchFrontNodeByBackendNode(const AnfNodePtr &backend_node, const K… in FetchFrontNodeByBackendNode()