Home
last modified time | relevance | path

Searched refs:anf_node_list (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/
Dtask_generator.h60 …bool GenTasks(const std::vector<CNodePtr> &anf_node_list, std::vector<TaskInfoPtr> *task_info_list,
70 …bool LaunchAllKernel(const std::vector<CNodePtr> &anf_node_list, std::vector<TaskInfoPtr> *task_in…
Dtask_generator.cc34 bool TaskGenerator::GenTasks(const std::vector<CNodePtr> &anf_node_list, std::vector<TaskInfoPtr> *… in GenTasks() argument
39 if (!LaunchAllKernel(anf_node_list, task_info_list, graph_id)) { in GenTasks()
254 bool TaskGenerator::LaunchAllKernel(const std::vector<CNodePtr> &anf_node_list, in LaunchAllKernel() argument
259 for (const auto &anf_node_ptr : anf_node_list) { in LaunchAllKernel()
/third_party/mindspore/tests/ut/cpp/stub/tasksink/
Dtask_sink_stub.cc26 bool TaskGenerator::GenTasks(const std::vector<CNodePtr> &anf_node_list, std::vector<TaskInfoPtr> *… in GenTasks() argument
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_kernel_runtime.cc475 auto anf_node_list = graph.execution_order(); in GenTask() local
477 if (!task_generator.GenTasks(anf_node_list, &task_info_list, graph.graph_id())) { in GenTask()