Commit Graph

4 Commits

Author SHA1 Message Date
Jack Zhou
391d66381f Remove eigen compliation option (#161)
Remove eigen option
2022-08-26 11:20:52 +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