| func.func @tensor() { |
| %0 = util.unfoldable_constant dense<[1.0, 2.0, 3.0, 4.0]> : tensor<4xf32> |
| %1 = util.unfoldable_constant dense<[5.0, 6.0, 7.0, 8.0]> : tensor<4xf32> |
| %result = "mhlo.add"(%0, %1) : (tensor<4xf32>, tensor<4xf32>) -> tensor<4xf32> |
| check.expect_almost_eq_const(%result, dense<[6.0, 8.0, 10.0, 12.0]> : tensor<4xf32>) : tensor<4xf32> |
| return |
| } |
| |
| func.func @tensor_4d() { |
| %0 = util.unfoldable_constant dense<[[[[1.0, 2.0], [3.0, 4.0]], |
| [[5.0, 6.0], [7.0, 8.0]]], |
| [[[9.0, 10.0], [11.0, 12.0]], |
| [[13.0, 14.0], [15.0, 16.0]]]]> : |
| tensor<2x2x2x2xf32> |
| %1 = util.unfoldable_constant dense<[[[[1.0, 2.0], [3.0, 4.0]], |
| [[5.0, 6.0], [7.0, 8.0]]], |
| [[[9.0, 10.0], [11.0, 12.0]], |
| [[13.0, 14.0], [15.0, 16.0]]]]> : |
| tensor<2x2x2x2xf32> |
| %result = "mhlo.add"(%0, %1) : (tensor<2x2x2x2xf32>, tensor<2x2x2x2xf32>) |
| -> tensor<2x2x2x2xf32> |
| check.expect_almost_eq_const(%result, dense<[[[[2.0, 4.0], [6.0, 8.0]], |
| [[10.0, 12.0], [14.0, 16.0]]], |
| [[[18.0, 20.0], [22.0, 24.0]], |
| [[26.0, 28.0], [30.0, 32.0]]]]> : |
| tensor<2x2x2x2xf32>) : tensor<2x2x2x2xf32> |
| return |
| } |
