huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-20 04:00:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

ONNX: Add Floor and Ceil (#2235)

Browse Source
This commit is contained in:
mokulus
2024-06-02 21:45:20 +02:00
committed by GitHub
parent 1ec3b2cc18
commit 03344d3c19
2 changed files with 162 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -877,6 +877,16 @@ pub fn simple_eval(
let output = input.relu()?;
values.insert(node.output[0].clone(), output);
}
"Ceil" => {
let input = get(&node.input[0])?;
let output = input.ceil()?;
values.insert(node.output[0].clone(), output);
}
"Floor" => {
let input = get(&node.input[0])?;
let output = input.floor()?;
values.insert(node.output[0].clone(), output);
}
// https://github.com/onnx/onnx/blob/main/docs/Operators.md#Constant
"Constant" => {
let value = match node.attribute.iter().find(|attr| attr.name == "value") {