Commit Graph

3 Commits

Author SHA1 Message Date
Jack Zhou
cec0d502e0 Add softmax function (#93)
* Add softmax function

* Add softmax unittest

* Add Softmax docs

* Add function directory

* Add comment for FD_VISIT_ALL_TYPES macro
2022-08-11 12:01:19 +08:00
Jack Zhou
7fb8dd7916 Add Transpose function (#91)
* Add Transpose function

* csrcs->csrc

* Add transpose unittest

* Add reduce_max_large_dim unittest
2022-08-10 19:00:16 +08:00
jiangjiajun
72eb130193 add csrc code 2022-08-10 02:52:36 +00:00