| func.func @no_round() { | |
| %0 = util.unfoldable_constant dense<[5, 8, -1, 7]> : tensor<4xi32> | |
| %1 = util.unfoldable_constant dense<[0, 1, 3, 1]> : tensor<4xi32> | |
| %result = tosa.arithmetic_right_shift %0, %1 round(false) : (tensor<4xi32>, tensor<4xi32>) -> tensor<4xi32> | |
| check.expect_eq_const(%result, dense<[5, 4, -1, 3]> : tensor<4xi32>) : tensor<4xi32> | |
| return | |
| } | |
| func.func @with_round() { | |
| %0 = util.unfoldable_constant dense<[5, 8, -1, 7]> : tensor<4xi32> | |
| %1 = util.unfoldable_constant dense<[0, 1, 3, 1]> : tensor<4xi32> | |
| %result = tosa.arithmetic_right_shift %0, %1 round(true) : (tensor<4xi32>, tensor<4xi32>) -> tensor<4xi32> | |
| check.expect_eq_const(%result, dense<[5, 4, 0, 4]> : tensor<4xi32>) : tensor<4xi32> | |
| return | |
| } |