FIX: bug fix

ucb-bar · Sep 5, 2024 · 3bc0c14 · 3bc0c14
1 parent 1b94c07
commit 3bc0c14
Showing 1 changed file with 31 additions and 31 deletions.
diff --git a/nn/nn.h b/nn/nn.h
@@ -5,37 +5,37 @@
 
 #include "float16.h"
 #include "tensor.h"
-#include "functional/nn_tensor_creation.h"
-#include "functional/nn_print.h"
-#include "functional/nn_abs.h"
-#include "functional/nn_add.h"
-#include "functional/nn_batch_norm2d.h"
-#include "functional/nn_clip.h"
-#include "functional/nn_conv2d.h"
-#include "functional/nn_copy.h"
-#include "functional/nn_div.h"
-#include "functional/nn_elu.h"
-#include "functional/nn_fill.h"
-#include "functional/nn_interpolate.h"
-#include "functional/nn_layer_norm.h"
-#include "functional/nn_linear.h"
-#include "functional/nn_matmul.h"
-#include "functional/nn_norm.h"
-#include "functional/nn_max.h"
-#include "functional/nn_mm.h"
-#include "functional/nn_maximum.h"
-#include "functional/nn_min.h"
-#include "functional/nn_minimum.h"
-#include "functional/nn_mul.h"
-#include "functional/nn_mv.h"
-#include "functional/nn_neg.h"
-#include "functional/nn_relu.h"
-#include "functional/nn_relu6.h"
-#include "functional/nn_rms_norm.h"
-#include "functional/nn_softmax.h"
-#include "functional/nn_silu.h"
-#include "functional/nn_sub.h"
-#include "functional/nn_sum.h"
+#include "functional/tensor_creation.h"
+#include "functional/print.h"
+#include "functional/abs.h"
+#include "functional/add.h"
+#include "functional/batch_norm2d.h"
+#include "functional/clip.h"
+#include "functional/conv2d.h"
+#include "functional/copy.h"
+#include "functional/div.h"
+#include "functional/elu.h"
+#include "functional/fill.h"
+#include "functional/interpolate.h"
+#include "functional/layer_norm.h"
+#include "functional/linear.h"
+#include "functional/matmul.h"
+#include "functional/norm.h"
+#include "functional/max.h"
+#include "functional/mm.h"
+#include "functional/maximum.h"
+#include "functional/min.h"
+#include "functional/minimum.h"
+#include "functional/mul.h"
+#include "functional/mv.h"
+#include "functional/neg.h"
+#include "functional/relu.h"
+#include "functional/relu6.h"
+#include "functional/rms_norm.h"
+#include "functional/softmax.h"
+#include "functional/silu.h"
+#include "functional/sub.h"
+#include "functional/sum.h"
 
 
 // http://elm-chan.org/junk/32bit/binclude.html