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