fix lstm

sonos · Aug 28, 2020 · e1c6499 · e1c6499
1 parent c76a76e
commit e1c6499
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/nnef/src/ops/core/scan.rs b/nnef/src/ops/core/scan.rs
@@ -129,6 +129,7 @@ fn ser_scan(ast: &mut IntoAst, node: &TypedNode) -> TractResult<Option<Arc<RValu
             ("full", array(full)),
             ("state", array(state)),
             ("output", array(outputs)),
+            ("skip", numeric(op.skip)),
         ],
     );
     ast.fragments.insert(body.decl.id.clone(), body);
@@ -217,6 +218,7 @@ fn de_scan(
             state: state.iter().any(|state| state.2 == output_name),
         });
     }
-    let op = Scan::new(body.model, input_mapping, output_mapping, None)?;
+    let skip:usize = invocation.named_arg_as(builder, "skip")?;
+    let op = Scan::new(body.model, input_mapping, output_mapping, Some(skip))?;
     builder.wire(op, &*outer_inputs)
 }