Searched refs:kAttrAtomicOutputIndexs (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_build_ascend.cc | 138 if (AnfAlgo::HasNodeAttr(kAttrAtomicOutputIndexs, pre_node)) { in CalCleanZerosSize()
139 … auto output_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicOutputIndexs); in CalCleanZerosSize()
254 if (AnfAlgo::HasNodeAttr(kAttrAtomicOutputIndexs, kernel_node)) { in IsAtomicNode()
255 output_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(kernel_node, kAttrAtomicOutputIndexs); in IsAtomicNode()
269 AnfAlgo::SetNodeAttr(kAttrAtomicOutputIndexs, MakeValue(output_indexs), kernel_node); in IsAtomicNode()
363 AnfAlgo::SetNodeAttr(kAttrAtomicOutputIndexs, MakeValue(indexes), anf_node); in TbeClearZeroNodeFusion()
432 AnfAlgo::SetNodeAttr(kAttrAtomicOutputIndexs, MakeValue(indexes), anf_node); in KernelBuildPreprocess()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.cc | 65 if (AnfAlgo::HasNodeAttr(kAttrAtomicOutputIndexs, anf_node_ptr)) { in LaunchAddrCleanAkgKernel()
66 …n_output_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(anf_node_ptr, kAttrAtomicOutputIndexs); in LaunchAddrCleanAkgKernel()
109 if (AnfAlgo::HasNodeAttr(kAttrAtomicOutputIndexs, pre_node)) { in LaunchAddrCleanKernel()
110 …clean_output_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicOutputIndexs); in LaunchAddrCleanKernel()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 1191 if (AnfAlgo::HasNodeAttr(kAttrAtomicOutputIndexs, pre_node)) { in GenAddrCleanLaunchArgs()
1193 …o clean_output_indexes = AnfAlgo::GetNodeAttr<std::vector<int>>(pre_node, kAttrAtomicOutputIndexs); in GenAddrCleanLaunchArgs()
1195 …lean_output_indexes = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicOutputIndexs); in GenAddrCleanLaunchArgs()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 359 constexpr auto kAttrAtomicOutputIndexs = "atomic_output_clean_indexs"; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/ |
| D | somas.cc | 598 if (AnfAlgo::HasNodeAttr(kAttrAtomicOutputIndexs, pre_node)) { in InitAtomicCleanInputs()
599 …clean_output_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicOutputIndexs); in InitAtomicCleanInputs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 2446 if (HasNodeAttr(kAttrAtomicOutputIndexs, pre_node)) { in CacheAddrForAtomicClean()
2448 auto clean_output_indexes = GetNodeAttr<std::vector<int>>(pre_node, kAttrAtomicOutputIndexs); in CacheAddrForAtomicClean()
2450 auto clean_output_indexes = GetNodeAttr<std::vector<size_t>>(pre_node, kAttrAtomicOutputIndexs); in CacheAddrForAtomicClean()
|