tests/e2e/tosa_ops/table.mlir

func.func @table() {
  %input = arith.constant dense<[-5405, 15214, -14896, 22008, 12529, -13501]> : tensor<6xi16>

  // This generates [0, ... 512] for a constant value to avoid an excessively large constant.
  %init = tensor.empty() : tensor<513xi16>
  %cst = linalg.generic {
    indexing_maps = [affine_map<(d0) -> (d0)>],
    iterator_types = ["parallel"]}
    outs(%init: tensor<513xi16>) {
    ^bb0(%arg1: i16):
      %i = linalg.index 0 : index
      %0 = arith.index_cast %i : index to i16
      linalg.yield %0 : i16
    } -> tensor<513xi16>

  %result = tosa.table %input, %cst : (tensor<6xi16>, tensor<513xi16>) -> tensor<6xi32>
  check.expect_eq_const(%result, dense<[27363, 47982, 17872, 54776, 45297, 19267]> : tensor<6xi32>) : tensor<6xi32>
  return
}