1ir_version: 7 2producer_name: "GlowONNXModelWriter" 3graph { 4 node { 5 input: "input" 6 output: "TopK" 7 output: "TopK:1" 8 name: "TopK" 9 op_type: "Glow_TopK" 10 attribute { 11 name: "o0_elemKind" 12 s: "i8" 13 type: STRING 14 } 15 attribute { 16 name: "o0_shape" 17 ints: 3 18 ints: 1 19 ints: 3 20 type: INTS 21 } 22 attribute { 23 name: "o0_qScale" 24 f: 1.2 25 type: FLOAT 26 } 27 attribute { 28 name: "o0_qOffset" 29 i: 5 30 type: INT 31 } 32 attribute { 33 name: "o1_elemKind" 34 s: "index64" 35 type: STRING 36 } 37 attribute { 38 name: "o1_shape" 39 ints: 3 40 ints: 1 41 ints: 3 42 type: INTS 43 } 44 attribute { 45 name: "K" 46 i: 3 47 type: INT 48 } 49 } 50 node { 51 input: "TopK:1" 52 output: "save_indices" 53 name: "save_indices_save" 54 op_type: "Identity" 55 } 56 node { 57 input: "TopK" 58 output: "save_values" 59 name: "save_values_save" 60 op_type: "Identity" 61 } 62 name: "glow" 63 input { 64 name: "input" 65 type { 66 tensor_type { 67 elem_type: 3 68 shape { 69 dim { 70 dim_value: 3 71 } 72 dim { 73 dim_value: 1 74 } 75 dim { 76 dim_value: 5 77 } 78 } 79 } 80 } 81 doc_string: "$offline:0$trainable:0$layout:*$elemKind:i8$qScale:1.20000004768372$qOffset:5" 82 } 83 output { 84 name: "save_values" 85 type { 86 tensor_type { 87 elem_type: 3 88 shape { 89 dim { 90 dim_value: 3 91 } 92 dim { 93 dim_value: 1 94 } 95 dim { 96 dim_value: 3 97 } 98 } 99 } 100 } 101 doc_string: "$offline:0$trainable:0$layout:*$elemKind:i8$qScale:1.20000004768372$qOffset:5$saveName:save_values_save" 102 } 103 output { 104 name: "save_indices" 105 type { 106 tensor_type { 107 elem_type: 7 108 shape { 109 dim { 110 dim_value: 3 111 } 112 dim { 113 dim_value: 1 114 } 115 dim { 116 dim_value: 3 117 } 118 } 119 } 120 } 121 doc_string: "$offline:0$trainable:0$layout:*$elemKind:index64$saveName:save_indices_save" 122 } 123} 124opset_import { 125 version: 9 126} 127