huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

add identity op (#1976)

Browse Source
This commit is contained in:
Mauro Sciancalepore
2024-04-01 12:08:25 +02:00
committed by GitHub
parent 318cb82f16
commit 888c09a3db
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
candle-onnx/src/eval.rs
View File

@ -776,6 +776,11 @@ pub fn simple_eval(
let output = input.reshape(new_shape)?; let output = input.reshape(new_shape)?;
values.insert(node.output[0].clone(), output); values.insert(node.output[0].clone(), output);
} }
// https://github.com/onnx/onnx/blob/main/docs/Operators.md#identity
"Identity" => {
let input = get(&node.input[0])?;
values.insert(node.output[0].clone(), input.clone());
}
op_type => bail!("unsupported op_type {op_type} for op {node:?}"), op_type => bail!("unsupported op_type {op_type} for op {node:?}"),
} }
} }