ADD: add tanh to header

ucb-bar · Oct 15, 2024 · 3ca85dc · 3ca85dc
1 parent b2ce84a
commit 3ca85dc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/nn/nn.h b/nn/nn.h
@@ -144,6 +144,11 @@ void nn_relu2d_f16(Tensor2D_F16 *y, const Tensor2D_F16 *x);
 void nn_relu2d_f32(Tensor2D_F32 *y, const Tensor2D_F32 *x);
 
 
+void nn_tanh2d_f16(Tensor2D_F16 *y, const Tensor2D_F16 *x);
+
+void nn_tanh2d_f32(Tensor2D_F32 *y, const Tensor2D_F32 *x);
+
+
 void nn_max1d_f16(Tensor0D_F16 *y, const Tensor1D_F16 *x);
 
 void nn_max1d_f32(Tensor0D_F32 *y, const Tensor1D_F32 *x);