pytorch

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