ack deprecation

sonos · May 28, 2024 · 5430882 · 5430882
1 parent 1e7201f
commit 5430882
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/harness/lstm-proptest-onnx-vs-tf/src/lib.rs b/harness/lstm-proptest-onnx-vs-tf/src/lib.rs
@@ -198,9 +198,11 @@ impl LstmProblem {
         let model = self.tf_model()?;
         let lstm_id = model.node_by_name("lstm")?.id;
         let memo_id = model.node_by_name("memo")?.id;
-        let plan_run = SimplePlan::new_for_outputs(
+        let plan_run = SimplePlan::build(
             &model,
             &[OutletId::new(lstm_id, 6), OutletId::new(memo_id, 0)],
+            &[],
+            &PlanOptions::default(),
         )?;
         let mut state = SimpleState::new(plan_run)?;
         let y = state.run(tvec!(self.x.clone()))?.remove(0);

diff --git a/harness/tf-moz-deepspeech/src/lib.rs b/harness/tf-moz-deepspeech/src/lib.rs
@@ -80,9 +80,11 @@ fn deepspeech_raw() -> TractResult<()> {
     let assign_2 = model.node_id_by_name("Assign_2")?;
     let assign_3 = model.node_id_by_name("Assign_3")?;
 
-    let plan = SimplePlan::new_for_outputs(
+    let plan = SimplePlan::build(
         model,
         &[logits.into(), (lstm, 1).into(), (lstm, 6).into(), assign_2.into(), assign_3.into()],
+        &[],
+        &PlanOptions::default(),
     )?;
 
     let mut state = SimpleState::new(plan)?;