pytorch

1
ir_version: 8
2
producer_name: "pytorch"
3
producer_version: "CURRENT_VERSION"
4
graph {
5
  node {
6
    output: "onnx::ConstantOfShape_11"
7
    name: "Constant_4"
8
    op_type: "Constant"
9
    attribute {
10
      name: "value"
11
      t {
12
        dims: 1
13
        data_type: 7
14
        raw_data: "\003\000\000\000\000\000\000\000"
15
      }
16
      type: TENSOR
17
    }
18
  }
19
  node {
20
    input: "onnx::ConstantOfShape_11"
21
    output: "onnx::Mul_3"
22
    name: "ConstantOfShape_5"
23
    op_type: "ConstantOfShape"
24
    attribute {
25
      name: "value"
26
      t {
27
        dims: 1
28
        data_type: 7
29
        raw_data: "\001\000\000\000\000\000\000\000"
30
      }
31
      type: TENSOR
32
    }
33
  }
34
  node {
35
    output: "onnx::Mul_4"
36
    name: "Constant_6"
37
    op_type: "Constant"
38
    attribute {
39
      name: "value"
40
      t {
41
        data_type: 7
42
        raw_data: "\377\377\377\377\377\377\377\377"
43
      }
44
      type: TENSOR
45
    }
46
  }
47
  node {
48
    input: "onnx::Mul_3"
49
    input: "onnx::Mul_4"
50
    output: "onnx::Equal_5"
51
    name: "Mul_7"
52
    op_type: "Mul"
53
  }
54
  node {
55
    output: "onnx::Equal_6"
56
    name: "Constant_8"
57
    op_type: "Constant"
58
    attribute {
59
      name: "value"
60
      t {
61
        dims: 3
62
        data_type: 7
63
        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"
64
      }
65
      type: TENSOR
66
    }
67
  }
68
  node {
69
    input: "onnx::Equal_6"
70
    input: "onnx::Equal_5"
71
    output: "onnx::Where_7"
72
    name: "Equal_9"
73
    op_type: "Equal"
74
  }
75
  node {
76
    output: "onnx::Where_8"
77
    name: "Constant_10"
78
    op_type: "Constant"
79
    attribute {
80
      name: "value"
81
      t {
82
        dims: 3
83
        data_type: 7
84
        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"
85
      }
86
      type: TENSOR
87
    }
88
  }
89
  node {
90
    input: "onnx::Where_7"
91
    input: "onnx::Mul_3"
92
    input: "onnx::Where_8"
93
    output: "onnx::Expand_9"
94
    name: "Where_11"
95
    op_type: "Where"
96
  }
97
  node {
98
    input: "onnx::Expand_0"
99
    input: "onnx::Expand_9"
100
    output: "10"
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: "10"
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