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 372c9cf commit b42c580Copy full SHA for b42c580
candle-onnx/src/eval.rs
@@ -358,6 +358,12 @@ fn simple_eval_(
358
let output = input0.broadcast_div(input1)?;
359
values.insert(node.output[0].clone(), output);
360
}
361
+ "Reciprocal" => {
362
+ let xs = get(&node.input[0])?;
363
+ let ones = Tensor::ones_like(&xs)?;
364
+ let output = ones.div(xs)?;
365
+ values.insert(node.output[0].clone(), output);
366
+ }
367
"Pow" => {
368
let input0 = get(&node.input[0])?;
369
let input1 = get(&node.input[1])?;
0 commit comments