pytorch
1ir_version: 8
2producer_name: "pytorch"
3producer_version: "CURRENT_VERSION"
4graph {
5node {
6input: "onnx::NonZero_0"
7output: "onnx::Transpose_1"
8name: "NonZero_0"
9op_type: "NonZero"
10}
11node {
12input: "onnx::Transpose_1"
13output: "2"
14name: "Transpose_1"
15op_type: "Transpose"
16attribute {
17name: "perm"
18ints: 1
19ints: 0
20type: INTS
21}
22}
23name: "main_graph"
24input {
25name: "onnx::NonZero_0"
26type {
27tensor_type {
28elem_type: 1
29shape {
30dim {
31dim_value: 2
32}
33dim {
34dim_value: 2
35}
36dim {
37dim_value: 2
38}
39}
40}
41}
42}
43output {
44name: "2"
45type {
46tensor_type {
47elem_type: 7
48shape {
49dim {
50dim_param: "Transpose2_dim_0"
51}
52dim {
53dim_value: 3
54}
55}
56}
57}
58}
59}
60opset_import {
61version: 17
62}
63