File size: 1,073 Bytes
0908a41
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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>>
  }
}