zhoushunjie
5147c1c750
Add fd tensor fp16
2022-10-04 02:44:24 +00:00
zhoushunjie
faf04b49a6
Add float16 tests
2022-10-02 09:21:00 +00:00
Jack Zhou
9c150f0bfb
Upgrade eigen func ( #253 )
...
* Add FDTensor copy and move assignment and constructor
* Upgrade the transpose to receive the output tensor same as input tensor
* Add note
* Add realloc for FDTensor
* Support output equals to input for softmax
* Remove FDTensor::Alloc
2022-09-20 10:58:07 +08:00
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
