summaryrefslogtreecommitdiff
path: root/candle-onnx/src/eval.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-onnx/src/eval.rs')
-rw-r--r--candle-onnx/src/eval.rs6
1 files changed, 6 insertions, 0 deletions
diff --git a/candle-onnx/src/eval.rs b/candle-onnx/src/eval.rs
index 358af7ac..2c60ed2f 100644
--- a/candle-onnx/src/eval.rs
+++ b/candle-onnx/src/eval.rs
@@ -1944,6 +1944,12 @@ fn simple_eval_(
values.insert(node.output[0].clone(), out);
}
+ // https://onnx.ai/onnx/operators/onnx__Sign.html
+ "Sign" => {
+ let input = get(&node.input[0])?;
+ let output = input.sign()?;
+ values.insert(node.output[0].clone(), output);
+ }
op_type => bail!("unsupported op_type {op_type} for op {node:?}"),
}
}