mirror of
https://github.com/zama-ai/concrete.git
synced 2026-04-17 03:00:54 -04:00
fix(frontend-python): support concatenating signed and unsigned tensors together
This commit is contained in:
@@ -647,7 +647,7 @@ class Context:
|
||||
)
|
||||
)
|
||||
x = self.encrypt(encrypted_type, x)
|
||||
sanitized_xs.append(x)
|
||||
sanitized_xs.append(self.to_signedness(x, of=resulting_type))
|
||||
|
||||
if axis is None:
|
||||
return self.operation(
|
||||
|
||||
@@ -171,6 +171,13 @@ from concrete import fhe
|
||||
"y": {"shape": (5, 2, 3)},
|
||||
},
|
||||
),
|
||||
pytest.param(
|
||||
lambda x, y: np.concatenate((x, y)),
|
||||
{
|
||||
"x": {"range": [0, 10], "shape": (4, 2)},
|
||||
"y": {"range": [-10, 10], "shape": (3, 2)},
|
||||
},
|
||||
),
|
||||
],
|
||||
)
|
||||
def test_concatenate(function, parameters, helpers):
|
||||
|
||||
Reference in New Issue
Block a user