summaryrefslogtreecommitdiff
path: root/candle-core/src/backprop.rs
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2023-09-21 06:19:10 +0100
committerGitHub <noreply@github.com>2023-09-21 06:19:10 +0100
commit7b26e513f15a0c7cd55ccfe48525bda1079427ce (patch)
tree6dabc58fb21552ddaf061c88406e82a4d3427ac3 /candle-core/src/backprop.rs
parentab1d40ea97b387f0dd05f77db37c840a4d624a08 (diff)
downloadcandle-7b26e513f15a0c7cd55ccfe48525bda1079427ce.tar.gz
candle-7b26e513f15a0c7cd55ccfe48525bda1079427ce.tar.bz2
candle-7b26e513f15a0c7cd55ccfe48525bda1079427ce.zip
Add the erf function. (#917)
Diffstat (limited to 'candle-core/src/backprop.rs')
-rw-r--r--candle-core/src/backprop.rs1
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-core/src/backprop.rs b/candle-core/src/backprop.rs
index 3e2ae1ed..a2548198 100644
--- a/candle-core/src/backprop.rs
+++ b/candle-core/src/backprop.rs
@@ -442,6 +442,7 @@ impl Tensor {
*sum_grad = sum_grad.add(&arg_grad)?
}
Op::Unary(_, UnaryOp::Gelu) => Err(Error::BackwardNotSupported { op: "gelu" })?,
+ Op::Unary(_, UnaryOp::Erf) => Err(Error::BackwardNotSupported { op: "erf" })?,
Op::Unary(_, UnaryOp::GeluErf) => {
Err(Error::BackwardNotSupported { op: "gelu-erf" })?
}