Searched refs:anf_node_ptr (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.cc | 60 void TaskGenerator::LaunchAddrCleanAkgKernel(const CNodePtr &anf_node_ptr, AddressPtrList *kernel_i… in LaunchAddrCleanAkgKernel() argument
61 MS_EXCEPTION_IF_NULL(anf_node_ptr); in LaunchAddrCleanAkgKernel()
65 if (AnfAlgo::HasNodeAttr(kAttrAtomicOutputIndexs, anf_node_ptr)) { in LaunchAddrCleanAkgKernel()
66 …auto clean_output_indexs = AnfAlgo::GetNodeAttr<std::vector<size_t>>(anf_node_ptr, kAttrAtomicOutp… in LaunchAddrCleanAkgKernel()
67 auto graph = anf_node_ptr->func_graph(); in LaunchAddrCleanAkgKernel()
72 if (node_users[anf_node_ptr].empty()) { in LaunchAddrCleanAkgKernel()
73 MS_LOG(EXCEPTION) << "Node users of " << anf_node_ptr->ToString() << " is empty."; in LaunchAddrCleanAkgKernel()
75 auto depend_node = node_users[anf_node_ptr].pop().first; in LaunchAddrCleanAkgKernel()
95 void TaskGenerator::LaunchAddrCleanKernel(const CNodePtr &anf_node_ptr, AddressPtrList *kernel_inpu… in LaunchAddrCleanKernel() argument
96 MS_EXCEPTION_IF_NULL(anf_node_ptr); in LaunchAddrCleanKernel()
[all …]
|
| D | task_generator.h | 67 static void LaunchAddrCleanKernel(const CNodePtr &anf_node_ptr, AddressPtrList *kernel_inputs);
68 static void LaunchAddrCleanAkgKernel(const CNodePtr &anf_node_ptr, AddressPtrList *kernel_inputs);
69 …bool LaunchKernel(const CNodePtr &anf_node_ptr, uint32_t stream_id, std::vector<TaskInfoPtr> *task…
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | generate_graph.cc | 88 auto anf_node_ptr = ValuePtrToAnfNodePtr(value_ptr); in CreateInt32Tensor() local
89 int_tensor_map[value] = anf_node_ptr; in CreateInt32Tensor()
90 return anf_node_ptr; in CreateInt32Tensor()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.h | 159 …eOutputsOfAnfNode(const FuncGraphPtr &kernel_graph, const AnfNodePtr &anf_node_ptr, size_t output_…
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 2070 bool IsNodeInputDynamicShape(const CNodePtr &anf_node_ptr) { in IsNodeInputDynamicShape() argument
2071 MS_EXCEPTION_IF_NULL(anf_node_ptr); in IsNodeInputDynamicShape()
2072 auto input_num = AnfAlgo::GetInputTensorNum(anf_node_ptr); in IsNodeInputDynamicShape()
2074 auto input_with_index = AnfAlgo::GetPrevNodeOutput(anf_node_ptr, i); in IsNodeInputDynamicShape()
2092 MS_LOG(INFO) << "Node:" << anf_node_ptr->fullname_with_scope() << "Invalid index:" << index in IsNodeInputDynamicShape()
|