Searched refs:none_index (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | add_placeholder_for_dynamic_gru.cc | 51 auto none_index = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(cnode, "placeholder_index"); in Process() local
53 for (size_t in_idx = 0; in_idx < input_num + none_index.size(); in_idx++) { in Process()
54 auto item = find(none_index.begin(), none_index.end(), in_idx); in Process()
55 if (item != none_index.end()) { in Process()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.cc | 175 … auto none_index = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(anf_node_ptr, "placeholder_index"); in LaunchKernel() local
176 auto item = find(none_index.begin(), none_index.end(), i); in LaunchKernel()
177 if (item != none_index.end()) { in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_adapter.cc | 428 auto none_index = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(node, "placeholder_index"); in IsPlaceHolderInput() local
429 return find(none_index.begin(), none_index.end(), input_ptr->index()) != none_index.end(); in IsPlaceHolderInput()
|
| D | tbe_kernel_build.cc | 327 auto none_index = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(anf_node, "placeholder_index"); in GenInputDescJson() local
328 auto item = find(none_index.begin(), none_index.end(), input_ptr->index()); in GenInputDescJson()
329 if (item != none_index.end()) { in GenInputDescJson()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 1143 auto none_index = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(cnode, "placeholder_index"); in GenLaunchArgs() local
1144 auto item = std::find(none_index.begin(), none_index.end(), i); in GenLaunchArgs()
1145 if (item != none_index.end()) { in GenLaunchArgs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 2395 auto none_index = GetNodeAttr<std::vector<int64_t>>(cnode, "placeholder_index"); in CacheAddrForKernel() local
2396 auto item = std::find(none_index.begin(), none_index.end(), i); in CacheAddrForKernel()
2397 if (item != none_index.end()) { in CacheAddrForKernel()
|