1ir_version: 8 2producer_name: "pytorch" 3producer_version: "CURRENT_VERSION" 4graph { 5 node { 6 input: "onnx::GatherElements_0" 7 input: "onnx::GatherElements_1" 8 output: "2" 9 name: "GatherElements_0" 10 op_type: "GatherElements" 11 attribute { 12 name: "axis" 13 i: 1 14 type: INT 15 } 16 } 17 name: "main_graph" 18 input { 19 name: "onnx::GatherElements_0" 20 type { 21 tensor_type { 22 elem_type: 1 23 shape { 24 dim { 25 dim_value: 3 26 } 27 dim { 28 dim_value: 4 29 } 30 dim { 31 dim_value: 3 32 } 33 } 34 } 35 } 36 } 37 input { 38 name: "onnx::GatherElements_1" 39 type { 40 tensor_type { 41 elem_type: 7 42 shape { 43 dim { 44 dim_value: 3 45 } 46 dim { 47 dim_value: 2 48 } 49 dim { 50 dim_value: 3 51 } 52 } 53 } 54 } 55 } 56 output { 57 name: "2" 58 type { 59 tensor_type { 60 elem_type: 1 61 shape { 62 dim { 63 dim_value: 3 64 } 65 dim { 66 dim_value: 2 67 } 68 dim { 69 dim_value: 3 70 } 71 } 72 } 73 } 74 } 75} 76opset_import { 77 version: 17 78} 79