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

dev(ir): add get_table function to ArbitraryFunction node

Browse Source
This commit is contained in:
Arthur Meyre
2021-08-30 16:16:35 +02:00
committed by Ayoub Benaissa
parent 74f0c9600e
commit dbda93639b
2 changed files with 27 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/common/extensions/test_table.py
View File

@@ -45,6 +45,8 @@ def test_lookup_table_encrypted_lookup(test_helpers):
x = EncryptedValue(Integer(2, is_signed=False))
op_graph = tracing.trace_numpy_function(f, {"x": x})
assert op_graph.output_nodes[0].get_table() == [3, 6, 0, 2]
ref_graph = nx.MultiDiGraph()
# Here is the ASCII drawing of the expected graph:
# (x) - (TLU)