Commit Graph

6 Commits

Author SHA1 Message Date
Jack Zhou
13a3d6f9f5 Upgrade FDASSERT string format (#121)
* upgrade FDASSERT

* Add c_str() for reduce

* upgrade FDASSERT flag

* upgrade flags

* upgrade lu

* fix tests cmake

* Fix pybind FDASSERT
2022-08-17 10:03:41 +08:00
Jack Zhou
b6247238f5 Add argmax, argmin function (#104)
* Add argmax argmin function

* Add unittest for argmax, argmin
2022-08-12 20:22:11 +08:00
Jack Zhou
5c6b3d6150 Upgrade FDASSERT (#97)
* upgrade FDASSERT

* Optimize error message of FDTensor function

* upgrade FDASSERT

Co-authored-by: Jason <jiangjiajun@baidu.com>
2022-08-11 19:42:31 +08:00
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