| 'use strict'; |
| |
| module.exports = { |
| tensor_dialect : $ => choice( |
| // operation ::= `tensor.empty` `(`$dynamicSizes`)` |
| // attr-dict `:` type($result) |
| seq('tensor.empty', |
| field('dynamicSizes', $._value_use_list_parens), |
| field('attributes', optional($.attribute)), |
| field('return', $._type_annotation)), |
| |
| // operation ::= `tensor.cast` $source attr-dict `:` |
| // type($source) `to` type($dest) |
| seq('tensor.cast', field('in', $.value_use), |
| field('attributes', optional($.attribute)), |
| field('return', $._type_annotation)), |
| |
| // operation ::= `tensor.dim` attr-dict $source `,` $index |
| // `:` type($source) |
| seq('tensor.dim', |
| field('attributes', optional($.attribute)), |
| field('tensor', $.value_use), ',', |
| field('index', $.value_use), |
| field('return', $._type_annotation)), |
| |
| // operation ::= `tensor.collapse_shape` $src |
| // $reassociation attr-dict `:` type($src) |
| // `into` type($result) |
| seq(choice('tensor.collapse_shape', 'tensor.expand_shape'), |
| field('tensor', $.value_use), |
| field('reassociation', $.nested_idx_list), |
| field('attributes', optional($.attribute)), |
| field('return', $._type_annotation)), |
| |
| // operation ::= `tensor.extract` $tensor `[` $indices `]` |
| // attr-dict `:` type($tensor) |
| seq('tensor.extract', field('tensor', $.value_use), |
| field('indices', $._dense_idx_list), |
| field('attributes', optional($.attribute)), |
| field('return', $._type_annotation)), |
| |
| // operation ::= `tensor.insert` $scalar `into` $dest `[` |
| // $indices `]` attr-dict |
| // `:` type($dest) |
| seq('tensor.insert', field('scalar', $.value_use), |
| token('into'), field('destination', $.value_use), |
| field('indices', $._dense_idx_list), |
| field('attributes', optional($.attribute)), |
| field('return', $._type_annotation)), |
| |
| // operation ::= `tensor.extract_slice` $source `` |
| // custom<DynamicIndexList>($offsets, |
| // $static_offsets) |
| // custom<DynamicIndexList>($sizes, |
| // $static_sizes) |
| // custom<DynamicIndexList>($strides, |
| // $static_strides) attr-dict `:` |
| // type($source) `to` type($result) |
| seq('tensor.extract_slice', field('tensor', $.value_use), |
| field('offsets', $._dense_idx_list), |
| field('sizes', $._dense_idx_list), |
| field('strides', $._dense_idx_list), |
| field('attributes', optional($.attribute)), |
| field('return', $._type_annotation)), |
| |
| // operation ::= `tensor.insert_slice` $source `into` $dest |
| // `` |
| // custom<DynamicIndexList>($offsets, |
| // $static_offsets) |
| // custom<DynamicIndexList>($sizes, |
| // $static_sizes) |
| // custom<DynamicIndexList>($strides, |
| // $static_strides) attr-dict `:` |
| // type($source) `into` type($dest) |
| // operation ::= `tensor.parallel_insert_slice` $source |
| // `into` $dest `` |
| // custom<DynamicIndexList>($offsets, |
| // $static_offsets) |
| // custom<DynamicIndexList>($sizes, |
| // $static_sizes) |
| // custom<DynamicIndexList>($strides, |
| // $static_strides) attr-dict `:` |
| // type($source) `into` type($dest) |
| seq(choice('tensor.insert_slice', |
| 'tensor.parallel_insert_slice'), |
| field('source', $.value_use), token('into'), |
| field('destination', $.value_use), |
| field('offsets', $._dense_idx_list), |
| field('sizes', $._dense_idx_list), |
| field('strides', $._dense_idx_list), |
| field('attributes', optional($.attribute)), |
| field('return', $._type_annotation)), |
| |
| // operation ::= `tensor.from_elements` $elements attr-dict |
| // `:` type($result) |
| seq('tensor.from_elements', |
| field('elements', optional($._value_use_list)), |
| field('attributes', optional($.attribute)), |
| field('return', $._type_annotation)), |
| |
| // operation ::= `tensor.gather` $source `[` $indices `]` |
| // `gather_dims` `(` $gather_dims `)` |
| // (`unique` $unique^)? |
| // attr-dict |
| // `:` functional-type(operands, results) |
| seq('tensor.gather', field('source', $.value_use), |
| field('indices', $._dense_idx_list), |
| field('gatherDims', $.gather_dims_attr), |
| field('unique', optional($.unique_attr)), |
| field('attributes', optional($.attribute)), |
| field('return', $._function_type_annotation)), |
| |
| // operation ::= `tensor.scatter` $source `into` $dest `[` |
| // $indices `]` |
| // `scatter_dims` `(` $scatter_dims `)` |
| // (`unique` $unique^)? |
| // attr-dict |
| // `:` functional-type(operands, results) |
| seq('tensor.scatter', field('source', $.value_use), |
| token('into'), field('destination', $.value_use), |
| field('indices', $._dense_idx_list), |
| field('scatterDims', $.scatter_dims_attr), |
| field('unique', optional($.unique_attr)), |
| field('attributes', optional($.attribute)), |
| field('return', $._function_type_annotation)), |
| |
| // operation ::= `tensor.pad` $source |
| // (`nofold` $nofold^)? |
| // `low` `` custom<DynamicIndexList>($low, |
| // $static_low) `high` `` |
| // custom<DynamicIndexList>($high, |
| // $static_high) $region attr-dict `:` |
| // type($source) `to` type($result) |
| seq('tensor.pad', field('source', $.value_use), |
| field('nofold', optional($.nofold_attr)), |
| field('low', seq(token('low'), $._dense_idx_list)), |
| field('high', seq(token('high'), $._dense_idx_list)), |
| field('body', $.region), |
| field('attributes', optional($.attribute)), |
| field('return', $._type_annotation)), |
| |
| // operation ::= `tensor.reshape` $source `(` $shape `)` |
| // attr-dict |
| // `:` functional-type(operands, results) |
| seq('tensor.reshape', field('tensor', $.value_use), |
| field('shape', $._value_use_list_parens), |
| field('attributes', optional($.attribute)), |
| field('return', $._function_type_annotation)), |
| |
| // operation ::= `tensor.splat` $input attr-dict `:` |
| // type($aggregate) |
| seq('tensor.splat', field('input', $.value_use), |
| field('attributes', optional($.attribute)), |
| field('return', $._type_annotation)), |
| |
| // operation ::= `tensor.pack` $source |
| // (`padding_value` `(` $padding_value^ `:` |
| // type($padding_value) `)`)? |
| // (`outer_dims_perm` `=` $outer_dims_perm^)? |
| // `inner_dims_pos` `=` $inner_dims_pos |
| // `inner_tiles` `=` |
| // custom<DynamicIndexList>($inner_tiles, |
| // $static_inner_tiles) `into` $dest |
| // attr-dict `:` type($source) `->` |
| // type($dest) |
| // operation ::= `tensor.unpack` $source |
| // (`outer_dims_perm` `=` $outer_dims_perm^)? |
| // `inner_dims_pos` `=` $inner_dims_pos |
| // `inner_tiles` `=` |
| // custom<DynamicIndexList>($inner_tiles, |
| // $static_inner_tiles) `into` $dest |
| // attr-dict `:` type($source) `->` |
| // type($dest) |
| seq(choice('tensor.pack', 'tensor.unpack'), |
| field('source', $.value_use), |
| field('padding_value', |
| optional(seq(token('padding_value'), '(', |
| $._value_use_and_type, ')'))), |
| field('outer_dims_perm', |
| optional($.outer_dims_perm_attr)), |
| field('inner_dims_pos', $.inner_dims_pos_attr), |
| field('inner_tiles', $.inner_tiles_attr), |
| token('into'), field('destination', $.value_use), |
| field('return', $._function_type_annotation)), |
| |
| // operation ::= `tensor.generate` $dynamicExtents $body |
| // attr-dict `:` type($result) |
| seq('tensor.generate', |
| field('dynamicExtents', $._value_use_list), |
| field('body', $.region), |
| field('attributes', optional($.attribute)), |
| field('return', $._type_annotation)), |
| |
| // operation ::= `tensor.rank` $tensor attr-dict `:` |
| // type($tensor) operation ::= `tensor.yield` $value |
| // attr-dict `:` type($value) |
| seq(choice('tensor.rank', 'tensor.yield'), |
| field('tensor', $.value_use), |
| field('attributes', optional($.attribute)), |
| field('return', $._type_annotation))) |
| } |
