fix docs and tests

gizatechxyz · Oct 23, 2023 · 19266ef · 19266ef
1 parent 6462c61
commit 19266ef
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/docs/framework/operators/neural-network/nn.thresholded_relu.md b/docs/framework/operators/neural-network/nn.thresholded_relu.md
@@ -1,7 +1,7 @@
 # NNTrait::thresholded_relu
 
 ```rust
- fn thresholded_relu(inputs: @Tensor<T>, alpha: @T) -> Tensor<T>
+ fn thresholded_relu(tensor: @Tensor<T>, alpha: @T) -> Tensor<T>
 ```
 
 Applies the thresholded rectified linear unit (Thresholded ReLU) activation function element-wise to a given tensor.
@@ -35,13 +35,13 @@ fn thresholded_relu_example() -> Tensor<FP8x23> {
             FixedTrait::new(0, false),
             FixedTrait::new(256, false),
             FixedTrait::new(512, false),
-            FixedTrait::new(513, false),
+            FixedTrait::new(257, false),
         ]
             .span(),
     );
     let alpha = FixedTrait::from_felt(256); // 1.0
 
     return NNTrait::leaky_relu(@tensor, @alpha);
 }
->>> [[0, 0], [512, 513]]
+>>> [[0, 0], [512, 257]]
 ```
diff --git a/nodegen/node/thresholded_relu.py b/nodegen/node/thresholded_relu.py
@@ -6,7 +6,7 @@
 class Thresholded_relu(RunAll):
 
     @staticmethod
-    def leaky_thresholded_fp8x23():
+    def thresholded_relu_fp8x23():
 
         alpha = 1.0
 
@@ -25,7 +25,7 @@ def leaky_thresholded_fp8x23():
                   name, Trait.NN)
 
     @staticmethod
-    def leaky_thresholded_fp16x16():
+    def thresholded_relu_fp16x16():
 
         alpha = 1.0
 

diff --git a/src/operators/nn/core.cairo b/src/operators/nn/core.cairo
@@ -454,7 +454,7 @@ trait NNTrait<T> {
     /// # NNTrait::thresholded_relu
     /// 
     /// ```rust
-    ///  fn thresholded_relu(inputs: @Tensor<T>, alpha: @T) -> Tensor<T>
+    ///  fn thresholded_relu(tensor: @Tensor<T>, alpha: @T) -> Tensor<T>
     /// ```
     ///
     /// Applies the thresholded rectified linear unit (Thresholded ReLU) activation function element-wise to a given tensor.
@@ -488,15 +488,15 @@ trait NNTrait<T> {
     ///             FixedTrait::new(0, false),
     ///             FixedTrait::new(256, false),
     ///             FixedTrait::new(512, false),
-    ///             FixedTrait::new(513, false),
+    ///             FixedTrait::new(257, false),
     ///         ]
     ///             .span(),
     ///     );
     ///     let alpha = FixedTrait::from_felt(256); // 1.0
     /// 
     ///     return NNTrait::leaky_relu(@tensor, @alpha);
     /// }
-    /// >>> [[0, 0], [512, 513]]
+    /// >>> [[0, 0], [512, 257]]
     /// ```
     /// 
     fn thresholded_relu(tensor: @Tensor<T>, alpha: @T) -> Tensor<T>;