Skip to content

Commit 9ed024b

Browse files
authored
Update onnx_subs.cpp
1 parent 6991a82 commit 9ed024b

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

app/Graph/onnx_subs.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,10 @@ using namespace it_lab_ai;
1414
int 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

Comments
 (0)