github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-09 12:15:09 -05:00
Code Issues Packages Projects Releases Wiki Activity

test: Add extract_slice_with_rank_reduction test since it is fixed (close #315)

Browse Source
This commit is contained in:
Quentin Bourgerie
2022-08-09 14:30:55 +02:00
parent cc6c2576ec
commit 58527a44c3
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
compiler/tests/end_to_end_fixture/end_to_end_encrypted_tensor.yaml
View File

@@ -75,6 +75,22 @@ tests:
- tensor: [ 5, 6, 7, 8, 9]
shape: [1,5]
---
description: extract_slice_with_rank_reduction
program: |
func.func @main(%t: tensor<2x10x!FHE.eint<6>>) -> tensor<5x!FHE.eint<6>> {
%r = tensor.extract_slice %t[1, 5][1, 5][1, 1] : tensor<2x10x!FHE.eint<6>> to tensor<5x!FHE.eint<6>>
return %r : tensor<5x!FHE.eint<6>>
}
tests:
- inputs:
- tensor: [63, 12, 7, 43, 52, 9, 26, 34, 22, 0,
0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
shape: [2,10]
width: 8
outputs:
- tensor: [ 5, 6, 7, 8, 9]
shape: [5]
---
description: extract_slice_stride
program: |
func.func @main(%t: tensor<2x10x!FHE.eint<6>>) -> tensor<1x5x!FHE.eint<6>> {