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