Home
last modified time | relevance | path

Searched defs:output_edges (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dir_converter.cc156 std::vector<EdgePtr> output_edges; in ConvertSingleOpOutputEdges() local
194 …std::vector<EdgePtr> output_edges = ConvertSingleOpOutputEdges(node, device_context, &address_to_e… in Convert() local
Dop_runner.cc563 void UpdateOutputShape(const std::vector<EdgePtr> &output_edges) { in UpdateOutputShape()
632 void AllocateOutputMemory(const std::vector<EdgePtr> &output_edges, const device::DeviceContext *de… in AllocateOutputMemory()
880 const auto &output_edges = single_op->outputs_; in RunSingleOpGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/graph_compiler/
Dgraph_partition.cc372 std::string ParallelSortDecideNextHandleTarget(const std::vector<AnfNodePtr> &output_edges, in ParallelSortDecideNextHandleTarget()
395 void ParallelSortVisitNodeEdges(const std::vector<AnfNodePtr> &output_edges, GraphNodesDependencyIn… in ParallelSortVisitNodeEdges()
456 auto &output_edges = output_edges_iter->second; in ParallelSort() local
Dbackend.cc233 const auto &output_edges = op_compiler_info->simple_graph_->outputs_; in GetOutputDeviceAddress() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Ddevice_address_utils.cc594 const auto &output_edges = op_compiler_info->simple_graph_->outputs_; in CreateGraphOutputDeviceAddress() local