@@ -14,9 +14,10 @@ using namespace it_lab_ai;
1414int main () {
1515 int type = 2 ;
1616 Tensor input = make_tensor (std::vector<int >({0 }));
17+ RuntimeOptions options;
1718 if (type == 0 ) {
1819 Graph graph1;
19- build_graph (graph1, input, input, MODEL_PATH_DENSENET_ONNX, false );
20+ build_graph (graph1, input, input, MODEL_PATH_DENSENET_ONNX, options, false );
2021
2122 Graph subgraph;
2223 Tensor scale = make_tensor (std::vector<float >({1.0 }));
@@ -65,7 +66,7 @@ int main() {
6566 std::cerr << " Time for concat:" << time2 << std::endl;
6667 } else if (type == 1 ) {
6768 Graph graph1;
68- build_graph (graph1, input, input, MODEL_PATH_RESNET_ONNX, false );
69+ build_graph (graph1, input, input, MODEL_PATH_RESNET_ONNX, options, false );
6970
7071 Graph subgraph;
7172 std::shared_ptr<Layer> layer_0 = std::make_shared<TransposeLayer>();
@@ -91,7 +92,7 @@ int main() {
9192 std::cerr << " Time for path5:" << time << std::endl;
9293 } else if (type == 2 ) {
9394 Graph graph1;
94- build_graph (graph1, input, input, MODEL_PATH_GOOGLENET_ONNX, false );
95+ build_graph (graph1, input, input, MODEL_PATH_GOOGLENET_ONNX, options, false );
9596
9697 Graph subgraph;
9798 Shape shape (2 );
0 commit comments