We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5305999 commit 136500eCopy full SHA for 136500e
crates/wasi-nn/src/wit.rs
@@ -249,12 +249,9 @@ impl generated::inference::HostGraphExecutionContext for WasiNnView<'_> {
249
tracing::debug!("compute with {} inputs", inputs.len());
250
251
let mut named_tensors = Vec::new();
252
- for (name, tensor_resopurce) in inputs.iter() {
253
- let tensor = self.table.get(&tensor_resopurce)?;
254
- named_tensors.push(crate::backend::NamedTensor {
255
- name: name.clone(),
256
- tensor: tensor.clone(),
257
- });
+ for (name, tensor_resopurce) in inputs.into_iter() {
+ let tensor = self.table.delete(tensor_resopurce)?;
+ named_tensors.push(crate::backend::NamedTensor { name, tensor });
258
}
259
260
let exec_context = &mut self.table.get_mut(&exec_context)?;
0 commit comments