Home
last modified time | relevance | path

Searched refs:output_used_num (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc998 std::vector<int64_t> output_used_num(output_num, 0); in GetNodeOutputUsedNum() local
1000 output_used_num[0] = SizeToLong(manager->node_users()[node].size()); in GetNodeOutputUsedNum()
1011 output_used_num[output_idx] = SizeToLong(manager->node_users()[out_getitem.first].size()); in GetNodeOutputUsedNum()
1014 return output_used_num; in GetNodeOutputUsedNum()
1018 auto output_used_num = GetNodeOutputUsedNum(kernel_graph, node); in GetNodeOutputTotalUsedNum() local
1019 return std::accumulate(output_used_num.begin(), output_used_num.end(), int64_t(0)); in GetNodeOutputTotalUsedNum()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dub_pattern_fusion.cc358 auto output_used_num = GetNodeOutputUsedNum(kernel_graph, node); in SetOutputUsedNumAttr() local
359 AnfAlgo::SetNodeAttr(kAttrOutputUsedNum, MakeValue(output_used_num), node); in SetOutputUsedNumAttr()