shortcut avoiding big salad of nodes

sonos · Sep 5, 2024 · 2898b88 · 2898b88
1 parent 6ceac7c
commit 2898b88
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/nnef/src/deser.rs b/nnef/src/deser.rs
@@ -669,6 +669,9 @@ impl CoerceFrom<Value> for OutletId {
  Value::Wire(outlet) => Ok(*outlet),
  Value::Tuple(tuple) if tuple.len() == 1 => OutletId::coerce(builder, &tuple[0]),
  Value::Array(inputs) => {
+ if let Ok(c) = from.to::<Arc<Tensor>>(builder) {
+ return builder.add_const(c)
+ }
  let mut outlets = tvec!();
  for i in inputs {
  let outlet = OutletId::coerce(builder, i)?;