1ir_version: 7 2producer_name: "pytorch" 3producer_version: "CURRENT_VERSION" 4graph { 5 node { 6 input: "input" 7 input: "target" 8 input: "weight" 9 output: "3" 10 name: "SoftmaxCrossEntropyLoss_0" 11 op_type: "SoftmaxCrossEntropyLoss" 12 attribute { 13 name: "ignore_index" 14 i: -100 15 type: INT 16 } 17 attribute { 18 name: "reduction" 19 s: "mean" 20 type: STRING 21 } 22 } 23 name: "main_graph" 24 initializer { 25 dims: 5 26 data_type: 1 27 name: "weight" 28 raw_data: "\334\204b?x\017\034\277C\300T?\246\205\346\275\227W\315\275" 29 } 30 input { 31 name: "input" 32 type { 33 tensor_type { 34 elem_type: 1 35 shape { 36 dim { 37 dim_value: 3 38 } 39 dim { 40 dim_value: 5 41 } 42 } 43 } 44 } 45 } 46 input { 47 name: "target" 48 type { 49 tensor_type { 50 elem_type: 7 51 shape { 52 dim { 53 dim_value: 3 54 } 55 } 56 } 57 } 58 } 59 output { 60 name: "3" 61 type { 62 tensor_type { 63 elem_type: 1 64 shape { 65 } 66 } 67 } 68 } 69} 70opset_import { 71 version: 12 72} 73