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