347 lines · plain
1[2 "ins"3 "outs"4 "else"5 "do"6 "loc"7 "attributes"8 "into"9 "to"10 "from"11 "step"12 "low"13 "high"14 "iter_args"15 "padding_value"16 "inner_tiles"17 "gather_dims"18 "scatter_dims"19 "outer_dims_perm"20 "inner_dims_pos"21 "shared_outs"22 "default"23 24 "eq"25 "ne"26 "oeq"27 "olt"28 "ole"29 "ogt"30 "oge"31 "slt"32 "sle"33 "sgt"34 "sge"35 "ult"36 "ule"37 "ugt"38 "uge"39] @keyword40 41[42 "module"43 "unrealized_conversion_cast"44 45 "func.call"46 "call"47 "func.call_indirect"48 "call_indirect"49 "func.constant"50 "constant"51 "func.func"52 "func.return"53 "return"54 55 "llvm.func"56 "llvm.return"57 58 "cf.assert"59 "cf.br"60 "cf.cond_br"61 "cf.switch"62 63 "scf.condition"64 "scf.execute_region"65 "scf.if"66 "scf.index_switch"67 "scf.for"68 "scf.forall"69 "scf.forall.in_parallel"70 "scf.parallel"71 "scf.reduce"72 "scf.reduce.return"73 "scf.while"74 "scf.yield"75 76 "arith.constant"77 "arith.addi"78 "arith.subi"79 "arith.divsi"80 "arith.divui"81 "arith.ceildivsi"82 "arith.ceildivui"83 "arith.floordivsi"84 "arith.remsi"85 "arith.remui"86 "arith.muli"87 "arith.mulsi_extended"88 "arith.mului_extended"89 "arith.andi"90 "arith.ori"91 "arith.xori"92 "arith.maxsi"93 "arith.maxui"94 "arith.minsi"95 "arith.minui"96 "arith.shli"97 "arith.shrsi"98 "arith.shrui"99 "arith.addui_extended"100 "arith.addf"101 "arith.divf"102 "arith.maximumf"103 "arith.minimumf"104 "arith.mulf"105 "arith.remf"106 "arith.subf"107 "arith.negf"108 "arith.cmpi"109 "arith.cmpf"110 "arith.extf"111 "arith.extsi"112 "arith.extui"113 "arith.fptosi"114 "arith.fptoui"115 "arith.index_cast"116 "arith.index_castui"117 "arith.sitofp"118 "arith.uitofp"119 "arith.bitcast"120 "arith.truncf"121 "arith.select"122 123 "math.absf"124 "math.atan"125 "math.cbrt"126 "math.ceil"127 "math.cos"128 "math.erf"129 "math.exp"130 "math.exp2"131 "math.expm1"132 "math.floor"133 "math.log"134 "math.log10"135 "math.log1p"136 "math.log2"137 "math.round"138 "math.roundeven"139 "math.rsqrt"140 "math.sin"141 "math.sqrt"142 "math.tan"143 "math.tanh"144 "math.trunc"145 "math.absi"146 "math.ctlz"147 "math.cttz"148 "math.ctpop"149 "math.atan2"150 "math.copysign"151 "math.fpowi"152 "math.powf"153 "math.ipowi"154 "math.fma"155 156 "memref.alloc"157 "memref.cast"158 "memref.copy"159 "memref.collapse_shape"160 "memref.expand_shape"161 "memref.prefetch"162 "memref.rank"163 "memref.realloc"164 "memref.view"165 166 "vector.bitcast"167 "vector.broadcast"168 "vector.shape_cast"169 "vector.type_cast"170 "vector.constant_mask"171 "vector.create_mask"172 "vector.extract"173 "vector.load"174 "vector.scalable.extract"175 "vector.fma"176 "vector.flat_transpose"177 "vector.insert"178 "vector.scalable.insert"179 "vector.shuffle"180 "vector.store"181 "vector.insert_strided_slice"182 "vector.matrix_multiply"183 "vector.print"184 "vector.transfer_read"185 "vector.transfer_write"186 "vector.yield"187 188 "tensor.empty"189 "tensor.cast"190 "tensor.dim"191 "tensor.collapse_shape"192 "tensor.expand_shape"193 "tensor.extract"194 "tensor.insert"195 "tensor.extract_slice"196 "tensor.insert_slice"197 "tensor.parallel_insert_slice"198 "tensor.from_elements"199 "tensor.gather"200 "tensor.scatter"201 "tensor.pad"202 "tensor.reshape"203 "tensor.splat"204 "tensor.pack"205 "tensor.unpack"206 "tensor.generate"207 "tensor.rank"208 "tensor.yield"209 210 "bufferization.alloc_tensor"211 "bufferization.to_buffer"212 "bufferization.to_tensor"213 214 "linalg.batch_matmul"215 "linalg.batch_matvec"216 "linalg.batch_reduce_matmul"217 "linalg.broadcast"218 "linalg.conv_1d_ncw_fcw"219 "linalg.conv_1d_nwc_wcf"220 "linalg.conv_1d"221 "linalg.conv_2d_nchw_fchw"222 "linalg.conv_2d_ngchw_fgchw"223 "linalg.conv_2d_nhwc_fhwc"224 "linalg.conv_2d_nhwc_hwcf"225 "linalg.conv_2d_nhwc_hwcf_q"226 "linalg.conv_2d"227 "linalg.conv_3d_ndhwc_dhwcf"228 "linalg.conv_3d_ndhwc_dhwcf_q"229 "linalg.conv_3d"230 "linalg.copy"231 "linalg.depthwise_conv_1d_nwc_wcm"232 "linalg.depthwise_conv_2d_nchw_chw"233 "linalg.depthwise_conv_2d_nhwc_hwc"234 "linalg.depthwise_conv_2d_nhwc_hwc_q"235 "linalg.depthwise_conv_2d_nhwc_hwcm"236 "linalg.depthwise_conv_2d_nhwc_hwcm_q"237 "linalg.depthwise_conv_3d_ndhwc_dhwc"238 "linalg.depthwise_conv_3d_ndhwc_dhwcm"239 "linalg.dot"240 "linalg.elemwise_binary"241 "linalg.elemwise_unary"242 "linalg.fill"243 "linalg.fill_rng_2d"244 "linalg.matmul"245 "linalg.matmul_unsigned"246 "linalg.matvec"247 "linalg.mmt4d"248 "linalg.pooling_nchw_max"249 "linalg.pooling_nchw_sum"250 "linalg.pooling_ncw_max"251 "linalg.pooling_ncw_sum"252 "linalg.pooling_ndhwc_max"253 "linalg.pooling_ndhwc_min"254 "linalg.pooling_ndhwc_sum"255 "linalg.pooling_nhwc_max"256 "linalg.pooling_nhwc_max_unsigned"257 "linalg.pooling_nhwc_min"258 "linalg.pooling_nhwc_min_unsigned"259 "linalg.pooling_nhwc_sum"260 "linalg.pooling_nwc_max"261 "linalg.pooling_nwc_max_unsigned"262 "linalg.pooling_nwc_min"263 "linalg.pooling_nwc_min_unsigned"264 "linalg.pooling_nwc_sum"265 "linalg.quantized_batch_matmul"266 "linalg.quantized_matmul"267 "linalg.vecmat"268 "linalg.generic"269 "linalg.index"270 "linalg.map"271 "linalg.yield"272] @function.builtin273 274(generic_operation) @function275 276(builtin_type) @type.builtin277 278[279 (type_alias)280 (dialect_type)281 (type_alias_def)282] @type283 284[285 (integer_literal)286 (float_literal)287 (complex_literal)288] @number289 290[291 (bool_literal)292 (tensor_literal)293 (array_literal)294 (unit_literal)295] @constant.builtin296 297(string_literal) @string298 299[300 (attribute_alias_def)301 (attribute_alias)302 (bare_attribute_entry)303 (attribute)304 (fastmath_attr)305 (scatter_dims_attr)306 (gather_dims_attr)307 (outer_dims_perm_attr)308 (inner_dims_pos_attr)309 (inner_tiles_attr)310 (unique_attr)311 (nofold_attr)312 (isWrite_attr)313 (localityHint_attr)314 (isDataCache_attr)315 (restrict_attr)316 (writable_attr)317] @attribute318 319[320 "("321 ")"322 "{"323 "}"324 "["325 "]"326] @punctuation.bracket327 328[329 ":"330 ","331] @punctuation.delimeter332 333[334 "="335 "->"336] @operator337 338(func_dialect name: (symbol_ref_id) @function)339(llvm_dialect name: (symbol_ref_id) @function)340 341(func_arg_list (value_use) @variable.parameter)342(block_arg_list (value_use) @variable.parameter)343 344(caret_id) @tag345(value_use) @variable346(comment) @comment347