minor bug fix

gizatechxyz · Oct 23, 2023 · 44700ab · 44700ab
1 parent 19266ef
commit 44700ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operators/nn/functional/thresholded_relu.cairo b/src/operators/nn/functional/thresholded_relu.cairo
@@ -22,7 +22,7 @@ fn thresholded_relu<
     loop {
         match z.data.pop_front() {
             Option::Some(item) => {
-                if (*item) < (*alpha) {
+                if (*item) <= (*alpha) {
                     data_result.append(NumberTrait::zero());
                 } else {
                     data_result.append(*item);