blob: 8a75b8e1d6df4d57f92b9ccbcd840b5e77baee20 [file] [log] [blame]
func @main(
%arg0: tensor<1x8x8x3xf32> {iree.identifier = "a"},
%arg1: tensor<1x8x8x3xf32> {iree.identifier = "b"},
%arg2: tensor<1x8x8x3xf32> {iree.identifier = "c"},
%arg3: tensor<1x8x8x3xf32> {iree.identifier = "d"}
) -> (
tensor<1x8x8x3xf32> {iree.identifier = "x"},
tensor<1x8x8x3xf32> {iree.identifier = "y"}
) {
%0 = "mhlo.add"(%arg1, %arg2) : (tensor<1x8x8x3xf32>, tensor<1x8x8x3xf32>) -> tensor<1x8x8x3xf32>
%1 = "mhlo.add"(%arg0, %0) : (tensor<1x8x8x3xf32>, tensor<1x8x8x3xf32>) -> tensor<1x8x8x3xf32>
%2 = "mhlo.add"(%arg3, %0) : (tensor<1x8x8x3xf32>, tensor<1x8x8x3xf32>) -> tensor<1x8x8x3xf32>
return %1, %2 : tensor<1x8x8x3xf32>, tensor<1x8x8x3xf32>
}