Commit Graph

4 Commits

Author SHA1 Message Date
heliqi
4d1f264d01 FDTensor support GPU device (#190)
* fdtensor support GPU

* TRT backend support GPU FDTensor

* FDHostAllocator add FASTDEPLOY_DECL

* fix FDTensor Data

* fix FDTensor dtype

Co-authored-by: Jason <jiangjiajun@baidu.com>
2022-09-08 16:53:08 +08:00
Jason
737b62a2ba Remove tensorrt/common codes (#171) 2022-08-29 19:21:18 +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