From e4f300b02752a87ea083069736fcc36999f67560 Mon Sep 17 00:00:00 2001
From: Mathieu Poumeyrol <mathieu.poumeyrol@sonos.com>
Date: Thu, 28 Nov 2024 15:57:50 +0100
Subject: [PATCH] use actual shapes for opaque weights case

---
 core/src/ops/einsum/mod.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/core/src/ops/einsum/mod.rs b/core/src/ops/einsum/mod.rs
index 4c969d2dac..8a78e4e9e9 100644
--- a/core/src/ops/einsum/mod.rs
+++ b/core/src/ops/einsum/mod.rs
@@ -295,7 +295,7 @@ impl TypedOp for EinSum {
     }
 
     fn cost(&self, inputs: &[&TypedFact]) -> TractResult<TVec<(Cost, TDim)>> {
-        let shapes: TVec<&[TDim]> = inputs.iter().map(|t| &*t.shape).collect();
+        let shapes = self.actual_input_shapes_from_facts(inputs)?;
         let oshape = eval::output_shape(&self.axes, &shapes)?;
         let ks = self
             .axes