diff --git a/harness/lstm-proptest-onnx-vs-tf/src/lib.rs b/harness/lstm-proptest-onnx-vs-tf/src/lib.rs index 5fc4d704f0..580880e20e 100644 --- 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 index 795f76b84d..63183f4e05 100644 --- 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)?;