From 893e9563db42f647981d169e8ac867c3e43ff6fc Mon Sep 17 00:00:00 2001 From: Luis Montero Date: Wed, 5 Jun 2024 10:01:29 +0200 Subject: [PATCH] chore: trying to fix cp's shape --- src/concrete/ml/common/preprocessors.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/concrete/ml/common/preprocessors.py b/src/concrete/ml/common/preprocessors.py index bb75bdadc..0b3434a7a 100644 --- a/src/concrete/ml/common/preprocessors.py +++ b/src/concrete/ml/common/preprocessors.py @@ -1571,12 +1571,12 @@ def add_extra_shape_to_subgraph(tlu_node, extra_dim: Tuple[int, ...]): node.evaluator.properties["constant"], node.evaluator.properties["constant"].shape + tuple([1 for _ in extra_dim]), ) + node.output.shape = node.evaluator.properties["constant"].shape continue node.inputs[0].shape = deepcopy(node.inputs[0].shape) node.inputs[0].shape = node.inputs[0].shape + extra_dim - # print(f"{node.output.shape=}") node.output.shape = deepcopy(node.output.shape) if node.output.shape[-1:] == extra_dim: # BIG HACK