Commit Graph

9 Commits

Author SHA1 Message Date
Jason
e93bf6e35c [Other] Add FDTensor function Pad (#532)
* Add InferShape func for all the vision processors

* fix infer shape of limit short

* Fix infer shape bug of stride_pad

* revert modify of processor

* add function pad
2022-11-08 21:45:31 +08:00
Jack Zhou
70f664161f [Functions] Add fd tensor concat (#507)
* Add fd tensor concat

* fix comment
2022-11-07 10:02:42 +08:00
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