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