Searched refs:ge_graph (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | graph_manager_test.cc | 46 std::shared_ptr<ge::Graph> ge_graph = std::make_shared<ge::Graph>(); in TEST_F() local
48 graph_manager.AddGraph("test_graph", ge_graph); in TEST_F()
52 ASSERT_EQ(ge_graph, wrappers.back()->graph_ptr_); in TEST_F()
56 ASSERT_EQ(ge_graph, wrappers2->graph_ptr_); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/ |
| D | op_tiling_adapter.cc | 149 … const tensor::TensorPtr &tensor_data, ge::ComputeGraphPtr *ge_graph, in NewConstantOp() argument
152 MS_EXCEPTION_IF_NULL(*ge_graph); in NewConstantOp()
177 ge::NodePtr constant_op = (*ge_graph)->AddNode(op_desc); in NewConstantOp()
185 ge::ComputeGraphPtr *ge_graph) { in ConvertDepends() argument
211 ge::NodePtr ge_constant_op = NewConstantOp(node, depend_name, const_tensor, ge_graph, index); in ConvertDepends()
270 …const CNodePtr &node, ge::ComputeGraphPtr *ge_graph, const std::map<uint32_t, tensor::TensorPtr> &… in AnfNodeToGeNodeAdapter() argument
284 ConvertDepends(node, depend_tensor_map, &op_desc, ge_graph); in AnfNodeToGeNodeAdapter()
285 MS_EXCEPTION_IF_NULL(*ge_graph); in AnfNodeToGeNodeAdapter()
286 auto ge_node = (*ge_graph)->AddNode(op_desc); in AnfNodeToGeNodeAdapter()
|
| D | op_tiling_adapter.h | 39 ge::NodePtr AnfNodeToGeNodeAdapter(const CNodePtr &node, ge::ComputeGraphPtr *ge_graph,
50 ge::ComputeGraphPtr *ge_graph);
52 ge::ComputeGraphPtr *ge_graph, size_t index);
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/ascend310/ |
| D | CMakeLists.txt | 3 find_library(ge_graph libgraph.so ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
10 target_link_libraries(ascend310_kernel_mid ${ge_graph} ${ge_compiler}
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | converter.cc | 210 ge::ComputeGraphPtr ge_graph = std::make_shared<ge::ComputeGraph>(kStubDataStructureName); in GenerateStubGeNode() local
211 MS_EXCEPTION_IF_NULL(ge_graph); in GenerateStubGeNode()
212 auto ge_node = ge_graph->AddNode(op_desc); in GenerateStubGeNode()
213 return {ge_node, ge_graph}; in GenerateStubGeNode()
|
| D | hccl_adapter.cc | 208 auto [ge_node, ge_graph] = GenerateStubGeNode(node, datatype); in GenTask()
233 ge_graph.reset(); in GenTask()
242 auto [ge_node, ge_graph] = GenerateStubGeNode(node, datatype); in CalcWorkspaceSize()
260 ge_graph.reset(); in CalcWorkspaceSize()
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/ |
| D | model_converter.cc | 111 transform::DfGraphPtr &ge_graph = wrap_ptr->graph_ptr_; in ConvertFuncGraphToAIR() local
112 if (ge_graph == nullptr) { in ConvertFuncGraphToAIR()
117 return ge_graph; in ConvertFuncGraphToAIR()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/ |
| D | ai_core_dynamic_kernel.cc | 139 ge::ComputeGraphPtr ge_graph = std::make_shared<ge::ComputeGraph>("default"); in ComputeTiling() local
140 …auto ge_node = converter.AnfNodeToGeNodeAdapter(cnode, &ge_graph, depend_tensor_map_, op_compile_i… in ComputeTiling()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline_ge.cc | 528 transform::DfGraphPtr ge_graph = wrap_ptr->graph_ptr_; in ExportDFGraph() local
529 if (ge_graph == nullptr) { in ExportDFGraph()
534 if (ge_graph->SaveToFile(file_name) != 0) { in ExportDFGraph()
|