Switch to shared_ptr in Graph class and change Graph interface

app/Accuracy/accuracy_check.cpp

@@ -47,29 +47,23 @@ int main() {
   }
   Tensor input = t;
   Tensor output = make_tensor(vec, sh1);
-  auto a1 = std::make_unique<InputLayer>(kNchw, kNchw, 1, 2);
-  Layer* a1_ptr = a1.get();
+  auto a1 = std::make_shared<InputLayer>(kNchw, kNchw, 1, 2);
   std::vector<float> kernelvec = {1, 1, 1, 1, 1, 1, 1, 1, 1};
   Shape sh2({3, 3});
   Tensor kernel = make_tensor(kernelvec, sh2);
-  auto a2 = std::make_unique<ConvolutionalLayer>(1, 0, 0, kernel);
-  Layer* a2_ptr = a2.get();
+  auto a2 = std::make_shared<ConvolutionalLayer>(1, 0, 0, kernel);
   Shape poolshape = {2, 2};
-  auto a3 = std::make_unique<EWLayer>("linear", 2.0F, 3.0F);
-  Layer* a3_ptr = a3.get();
-  auto a4 = std::make_unique<PoolingLayer>(poolshape, "average");
-  Layer* a4_ptr = a4.get();
-  auto a5 = std::make_unique<OutputLayer>();
-  Layer* a5_ptr = a5.get();
-  auto a6 = std::make_unique<FCLayer>();
-  Layer* a6_ptr = a6.get();
-  graph.setInput(a1_ptr, input);
-  graph.makeConnection(a1_ptr, a2_ptr);
-  graph.makeConnection(a2_ptr, a3_ptr);
-  graph.makeConnection(a3_ptr, a4_ptr);
-  graph.makeConnection(a4_ptr, a5_ptr);
-  graph.makeConnection(a5_ptr, a6_ptr);
-  graph.setOutput(a5_ptr, output);
+  auto a3 = std::make_shared<EWLayer>("linear", 2.0F, 3.0F);
+  auto a4 = std::make_shared<PoolingLayer>(poolshape, "average");
+  auto a5 = std::make_shared<OutputLayer>();
+  auto a6 = std::make_shared<FCLayer>();
+  graph.setInput(a1, input);
+  graph.makeConnection(a1, a2);
+  graph.makeConnection(a2, a3);
+  graph.makeConnection(a3, a4);
+  graph.makeConnection(a4, a5);
+  graph.makeConnection(a5, a6);
+  graph.setOutput(a5, output);
   graph.inference(options);
   std::vector<float> tmp = *output.as<float>();
   std::vector<float> tmp_output = softmax<float>(*output.as<float>());

app/Graph/CMakeLists.txt

@@ -22,6 +22,11 @@ target_link_libraries(Graph_Build BuildGraph)
 add_executable(ACC acc_check.cpp)
 target_link_libraries(ACC BuildGraph)
 
+add_executable(onnx_subgraphs onnx_subgraphs.cpp)
+target_link_libraries(onnx_subgraphs BuildGraph)
+target_link_libraries(onnx_subgraphs OpenMP::OpenMP_CXX)
+target_link_libraries(onnx_subgraphs graphT_lib)
+
 file(DOWNLOAD
     "https://raw.githubusercontent.com/DeepTrackAI/MNIST_dataset/main/mnist/test/1_000008.png"
     "${CMAKE_SOURCE_DIR}/docs/input/28/test1.png"