Home
last modified time | relevance | path

Searched refs:output_item (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_memory_manager.cc139 for (auto &output_item : summary_outputs) { in IncreaseSummaryRefCount() local
140 auto node = output_item.second.first; in IncreaseSummaryRefCount()
141 size_t index = IntToSize(output_item.second.second); in IncreaseSummaryRefCount()
155 for (auto &output_item : summary_outputs) { in DecreaseSummaryRefCount() local
156 auto node = output_item.second.first; in DecreaseSummaryRefCount()
157 size_t index = IntToSize(output_item.second.second); in DecreaseSummaryRefCount()
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dspliter.cc47 for (const auto &output_item : nodes_inputs_) { in VisitNodesOutputs() local
48 if (output_item.first != node) { in VisitNodesOutputs()
49 for (const auto &output : output_item.second) { in VisitNodesOutputs()
51 nodes_outputs_[node].insert(output_item.first); in VisitNodesOutputs()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_select.cc154 SupportFormatItem output_item; in GetAgnosticPatternKernelInfo() local
156 output_item.assign(op_info.outputs_ptr().size(), format); in GetAgnosticPatternKernelInfo()
158 support_format.output_format.emplace_back(output_item); in GetAgnosticPatternKernelInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dub_pattern_fusion.cc168 int64_t fusion_id, const AnfNodePtr &output_item, in ReplaceInputNodeInOtherFusionScope() argument
172 output_item); in ReplaceInputNodeInOtherFusionScope()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc1867 for (auto &output_item : summary_outputs) { in Summary() local
1868 auto node = output_item.second.first; in Summary()
1869 size_t index = IntToSize(output_item.second.second); in Summary()
1885 params_list[output_item.first] = tensor; in Summary()