| 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 |
| } |
