blob: c4f3ca5923fb44d9beb47f10b7a09cf10438cfbd [file]
func.func @pw_add() {
%0 = util.unfoldable_constant dense<[[1, 2, 3, 4, 5, 6, 7, 8],
[9, 10, 11, 12, 13, 14, 15, 16],
[17, 18, 19, 20, 21, 22, 23, 24],
[25, 26, 27, 28, 29, 30, 31, 32]]> : tensor<4x8xi32>
%1 = util.unfoldable_constant dense<[[2, 4, 6, 8, 10, 12, 14, 16],
[18, 20, 22, 24, 26, 28, 30, 32],
[34, 36, 38, 40, 42, 44, 46, 48],
[50, 52, 54, 56, 58, 60, 62, 64]]> : tensor<4x8xi32>
%2 = "mhlo.add"(%0, %1) : (tensor<4x8xi32>, tensor<4x8xi32>) -> tensor<4x8xi32>
check.expect_eq_const(%2, dense<[[3, 6, 9, 12, 15, 18, 21, 24],
[27, 30, 33, 36, 39, 42, 45, 48],
[51, 54, 57, 60, 63, 66, 69, 72],
[75, 78, 81, 84, 87, 90, 93, 96]]> : tensor<4x8xi32>) : tensor<4x8xi32>
return
}