use actual shapes for opaque weights case

sonos · Nov 28, 2024 · 19fe42c · 19fe42c
1 parent 69b5aab
commit 19fe42c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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