| Thomas | f0f64ca | 2022-03-30 10:38:01 -0700 | [diff] [blame] | 1 | func.func @tensor_float() { |
| Ben Vanik | a1ff64e | 2021-08-06 10:13:35 -0700 | [diff] [blame] | 2 | %0 = util.unfoldable_constant dense<[1.0, -1.5, 7.0, -2.0]> : tensor<4xf32> |
| 3 | %1 = util.unfoldable_constant dense<[5.0, 1.0, 7.0, -3.0]> : tensor<4xf32> |
| Han-Chung Wang | 42e54ab | 2023-08-22 23:10:14 -0700 | [diff] [blame] | 4 | %result = tosa.maximum %0, %1 : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32> |
| NatashaKnk | a89c9d8 | 2021-02-04 15:25:33 -0800 | [diff] [blame] | 5 | check.expect_eq_const(%result, dense<[5.0, 1.0, 7.0, -2.0]> : tensor<4xf32>) : tensor<4xf32> |
| 6 | return |
| 7 | } |
| 8 | |
| Thomas | f0f64ca | 2022-03-30 10:38:01 -0700 | [diff] [blame] | 9 | func.func @tensor_int() { |
| Ben Vanik | a1ff64e | 2021-08-06 10:13:35 -0700 | [diff] [blame] | 10 | %0 = util.unfoldable_constant dense<[1, 0, 5, 3]> : tensor<4xi32> |
| 11 | %1 = util.unfoldable_constant dense<[5, 0, 1, -8]> : tensor<4xi32> |
| Han-Chung Wang | 42e54ab | 2023-08-22 23:10:14 -0700 | [diff] [blame] | 12 | %result = tosa.maximum %0, %1 : (tensor<4xi32>, tensor<4xi32>) -> tensor<4xi32> |
| NatashaKnk | a89c9d8 | 2021-02-04 15:25:33 -0800 | [diff] [blame] | 13 | check.expect_eq_const(%result, dense<[5, 0, 5, 3]> : tensor<4xi32>) : tensor<4xi32> |
| 14 | return |
| 15 | } |
