pytorch

1
ir_version: 8
2
producer_name: "pytorch"
3
producer_version: "CURRENT_VERSION"
4
graph {
5
  node {
6
    output: "onnx::Where_1"
7
    name: "Constant_4"
8
    op_type: "Constant"
9
    attribute {
10
      name: "value"
11
      t {
12
        dims: 3
13
        data_type: 7
14
        raw_data: "\004\000\000\000\000\000\000\000\006\000\000\000\000\000\000\000\002\000\000\000\000\000\000\000"
15
      }
16
      type: TENSOR
17
    }
18
  }
19
  node {
20
    output: "onnx::ConstantOfShape_10"
21
    name: "Constant_5"
22
    op_type: "Constant"
23
    attribute {
24
      name: "value"
25
      t {
26
        dims: 1
27
        data_type: 7
28
        raw_data: "\003\000\000\000\000\000\000\000"
29
      }
30
      type: TENSOR
31
    }
32
  }
33
  node {
34
    input: "onnx::ConstantOfShape_10"
35
    output: "onnx::Mul_3"
36
    name: "ConstantOfShape_6"
37
    op_type: "ConstantOfShape"
38
    attribute {
39
      name: "value"
40
      t {
41
        dims: 1
42
        data_type: 7
43
        raw_data: "\001\000\000\000\000\000\000\000"
44
      }
45
      type: TENSOR
46
    }
47
  }
48
  node {
49
    output: "onnx::Mul_4"
50
    name: "Constant_7"
51
    op_type: "Constant"
52
    attribute {
53
      name: "value"
54
      t {
55
        data_type: 7
56
        raw_data: "\377\377\377\377\377\377\377\377"
57
      }
58
      type: TENSOR
59
    }
60
  }
61
  node {
62
    input: "onnx::Mul_3"
63
    input: "onnx::Mul_4"
64
    output: "onnx::Equal_5"
65
    name: "Mul_8"
66
    op_type: "Mul"
67
  }
68
  node {
69
    output: "onnx::Equal_6"
70
    name: "Constant_9"
71
    op_type: "Constant"
72
    attribute {
73
      name: "value"
74
      t {
75
        dims: 3
76
        data_type: 7
77
        raw_data: "\004\000\000\000\000\000\000\000\006\000\000\000\000\000\000\000\002\000\000\000\000\000\000\000"
78
      }
79
      type: TENSOR
80
    }
81
  }
82
  node {
83
    input: "onnx::Equal_6"
84
    input: "onnx::Equal_5"
85
    output: "onnx::Where_7"
86
    name: "Equal_10"
87
    op_type: "Equal"
88
  }
89
  node {
90
    input: "onnx::Where_7"
91
    input: "onnx::Mul_3"
92
    input: "onnx::Where_1"
93
    output: "onnx::Expand_8"
94
    name: "Where_11"
95
    op_type: "Where"
96
  }
97
  node {
98
    input: "onnx::Expand_0"
99
    input: "onnx::Expand_8"
100
    output: "9"
101
    name: "Expand_12"
102
    op_type: "Expand"
103
  }
104
  name: "main_graph"
105
  input {
106
    name: "onnx::Expand_0"
107
    type {
108
      tensor_type {
109
        elem_type: 1
110
        shape {
111
          dim {
112
            dim_value: 6
113
          }
114
          dim {
115
            dim_value: 1
116
          }
117
        }
118
      }
119
    }
120
  }
121
  output {
122
    name: "9"
123
    type {
124
      tensor_type {
125
        elem_type: 1
126
        shape {
127
          dim {
128
            dim_value: 4
129
          }
130
          dim {
131
            dim_value: 6
132
          }
133
          dim {
134
            dim_value: 2
135
          }
136
        }
137
      }
138
    }
139
  }
140
}
141
opset_import {
142
  version: 17
143
}
144