File tree Expand file tree Collapse file tree 2 files changed +7
-5
lines changed
samples/sampleOnnxMnistCoordConvAC Expand file tree Collapse file tree 2 files changed +7
-5
lines changed Original file line number Diff line number Diff line change 1- onnx == 1.9.0 ; (platform_machine=="x86_64" and sys.platform=="linux" )
2- onnx == 1.10.0 ; (platform_machine=="aarch64" and sys.platform=="linux" )
1+ onnx == 1.10.2
32tensorflow-gpu == 2.5.1 ; (platform_machine=="x86_64" and sys.platform=="linux" )
43onnxruntime == 1.8.1
54-f https://download.pytorch.org/whl/cu113/torch_stable.html
65torch == 1.10.2+cu113
6+ torchvision == 0.11.3
77Pillow == 8.3.2
88numpy
99pycuda < 2021.1
1010pytest
11+ --extra-index-url https://pypi.ngc.nvidia.com
12+ onnx-graphsurgeon
Original file line number Diff line number Diff line change @@ -57,11 +57,11 @@ def main():
5757 tmap = graph .tensors ()
5858 # You can figure out the input and output tensors using Netron.
5959 inputs = [tmap ["conv1" ]]
60- outputs = [tmap ["input " ]]
60+ outputs = [tmap ["90 " ]]
6161 replace_with_coordconvac (graph , inputs , outputs )
6262
63- inputs = [tmap ["onnx::Cast_90 " ]]
64- outputs = [tmap ["input.3 " ]]
63+ inputs = [tmap ["92 " ]]
64+ outputs = [tmap ["170 " ]]
6565 replace_with_coordconvac (graph , inputs , outputs )
6666
6767 # Remove the now-dangling subgraph.
You can’t perform that action at this time.
0 commit comments