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 dim { 36 dim_value: 2 37 } 38 dim { 39 dim_value: 1 40 } 41 } 42 } 43 } 44 } 45 input { 46 name: "target" 47 type { 48 tensor_type { 49 elem_type: 7 50 shape { 51 dim { 52 dim_value: 3 53 } 54 dim { 55 dim_value: 2 56 } 57 dim { 58 dim_value: 1 59 } 60 } 61 } 62 } 63 } 64 output { 65 name: "2" 66 type { 67 tensor_type { 68 elem_type: 1 69 shape { 70 } 71 } 72 } 73 } 74} 75opset_import { 76 version: 12 77} 78