diff --git a/crates/wasi-nn/src/wit.rs b/crates/wasi-nn/src/wit.rs index a5cd9812bfa4..48703e3eb492 100644 --- a/crates/wasi-nn/src/wit.rs +++ b/crates/wasi-nn/src/wit.rs @@ -249,12 +249,9 @@ impl generated::inference::HostGraphExecutionContext for WasiNnView<'_> { tracing::debug!("compute with {} inputs", inputs.len()); let mut named_tensors = Vec::new(); - for (name, tensor_resopurce) in inputs.iter() { - let tensor = self.table.get(&tensor_resopurce)?; - named_tensors.push(crate::backend::NamedTensor { - name: name.clone(), - tensor: tensor.clone(), - }); + for (name, tensor_resopurce) in inputs.into_iter() { + let tensor = self.table.delete(tensor_resopurce)?; + named_tensors.push(crate::backend::NamedTensor { name, tensor }); } let exec_context = &mut self.table.get_mut(&exec_context)?;