Home
last modified time | relevance | path

Searched refs:summary_nodes (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_reuse.cc372 auto summary_nodes = graph_->summary_nodes(); in SetSummaryNodesRefCount() local
373 if (summary_nodes.empty()) { in SetSummaryNodesRefCount()
378 for (auto &node_item : summary_nodes) { in SetSummaryNodesRefCount()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc264 const std::map<std::string, std::pair<AnfNodePtr, int>> &summary_nodes = graph->summary_nodes(); in SetSummaryNodesRefCount() local
265 if (summary_nodes.empty()) { in SetSummaryNodesRefCount()
269 for (const auto &item : summary_nodes) { in SetSummaryNodesRefCount()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_session.cc1626 auto summary_nodes = graph->summary_nodes(); in RecurseSetSummaryNodes() local
1627 summary->insert(summary_nodes.begin(), summary_nodes.end()); in RecurseSetSummaryNodes()
1638 auto child_graph_summary = child_graph->summary_nodes(); in RecurseSetSummaryNodes()
1648 auto summary_nodes = graph->summary_nodes(); in SetSummaryNodes() local
1650 summary.insert(summary_nodes.begin(), summary_nodes.end()); in SetSummaryNodes()
Dcpu_session.cc153 runtime_.IncreaseSummaryRefCount(graph->summary_nodes()); in CompileGraphImpl()
Dkernel_graph.h225 …const std::map<std::string, std::pair<AnfNodePtr, int>> &summary_nodes() const { return summary_no… in summary_nodes() function
Dsession_basic.cc1822 auto summary = graph->summary_nodes(); in SetSummaryNodes()
1864 auto summary_outputs = graph->summary_nodes(); in Summary()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/
Dsomas.cc756 auto summary_nodes = graph->summary_nodes(); in SummaryInputProcess() local
757 if (summary_nodes.empty()) { in SummaryInputProcess()
762 for (auto &node_item : summary_nodes) { in SummaryInputProcess()