Searched refs:ge_node (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | hccl_adapter.cc | 208 auto [ge_node, ge_graph] = GenerateStubGeNode(node, datatype); in GenTask()
209 MS_EXCEPTION_IF_NULL(ge_node); in GenTask()
210 auto op = ge_node->GetOpDesc(); in GenTask()
215 ge::Status ret = ops_kernel_builder_->CalcOpRunningParam(*ge_node); in GenTask()
223 ret = ops_kernel_builder_->GenerateTask(*ge_node, unused_ctx, domi_tasks); in GenTask()
242 auto [ge_node, ge_graph] = GenerateStubGeNode(node, datatype); in CalcWorkspaceSize()
243 MS_EXCEPTION_IF_NULL(ge_node); in CalcWorkspaceSize()
244 auto op = ge_node->GetOpDesc(); in CalcWorkspaceSize()
248 ge::Status ret = ops_kernel_builder_->CalcOpRunningParam(*ge_node); in CalcWorkspaceSize()
|
| D | converter.cc | 212 auto ge_node = ge_graph->AddNode(op_desc); in GenerateStubGeNode() local
213 return {ge_node, ge_graph}; in GenerateStubGeNode()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/ |
| D | op_tiling_adapter.cc | 220 void OpTilingCalculateAdapter::AddEdge(const ge::NodePtr &ge_node, in AddEdge() argument
222 MS_EXCEPTION_IF_NULL(ge_node); in AddEdge()
227 (void)ge_node->AddLinkFrom(index, constant_op); in AddEdge()
286 auto ge_node = (*ge_graph)->AddNode(op_desc); in AnfNodeToGeNodeAdapter() local
287 MS_EXCEPTION_IF_NULL(ge_node); in AnfNodeToGeNodeAdapter()
288 AddEdge(ge_node, constant_ops); in AnfNodeToGeNodeAdapter()
289 return ge_node; in AnfNodeToGeNodeAdapter()
|
| D | op_tiling_adapter.h | 53 …void AddEdge(const ge::NodePtr &ge_node, const std::vector<std::tuple<std::size_t, ge::NodePtr>> &…
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | ai_core_dynamic_kernel.cc | 140 …auto ge_node = converter.AnfNodeToGeNodeAdapter(cnode, &ge_graph, depend_tensor_map_, op_compile_i… in ComputeTiling() local
141 (void)optiling::OpParaCalculateV2(*ge_node, op_run_info_v2); in ComputeTiling()
|