module { | |
func.func @main(%arg0: tensor<1x!FHE.eint<4>>, %arg1: tensor<1x!FHE.eint<4>>) -> tensor<1x!FHE.eint<4>> { | |
%0 = "FHELinalg.sum"(%arg0) {axes = [], keep_dims = false} : (tensor<1x!FHE.eint<4>>) -> !FHE.eint<4> | |
%1 = "FHELinalg.sum"(%arg1) {axes = [], keep_dims = false} : (tensor<1x!FHE.eint<4>>) -> !FHE.eint<4> | |
%cst = arith.constant dense<1> : tensor<1xi2> | |
%from_elements = tensor.from_elements %0 : tensor<1x!FHE.eint<4>> | |
%2 = "FHELinalg.add_eint_int"(%from_elements, %cst) : (tensor<1x!FHE.eint<4>>, tensor<1xi2>) -> tensor<1x!FHE.eint<4>> | |
%3 = "FHELinalg.sub_eint"(%2, %from_elements) : (tensor<1x!FHE.eint<4>>, tensor<1x!FHE.eint<4>>) -> tensor<1x!FHE.eint<4>> | |
%from_elements_0 = tensor.from_elements %1 : tensor<1x!FHE.eint<4>> | |
%4 = "FHELinalg.add_eint"(%3, %from_elements_0) : (tensor<1x!FHE.eint<4>>, tensor<1x!FHE.eint<4>>) -> tensor<1x!FHE.eint<4>> | |
%5 = "FHELinalg.sub_eint"(%4, %from_elements_0) : (tensor<1x!FHE.eint<4>>, tensor<1x!FHE.eint<4>>) -> tensor<1x!FHE.eint<4>> | |
return %5 : tensor<1x!FHE.eint<4>> | |
} | |
} |