OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
10,793 Commits 8,304 Branches 150 Tags 1.5 GiB
7f587de4bc
Commit Graph

3 Commits

Author SHA1 Message Date
Yinghai Lu
7f587de4bc
[Caffe2] Let TensorRT flow use the generic graph transformer (#6696) 
* Refine the transform API

* Let TensorRT flow use the generic graph transformer

* Rebase
 2018-04-19 10:07:01 -07:00
Yinghai Lu
6252706feb
[Caffe2] Workspace centric API for TensorRT transformation (#6678) 
* Workspace centric API for trt transformation

* Merge SSA rewrite code
 2018-04-17 21:23:27 -07:00
Yinghai Lu
434f710f3f
[Caffe2] Add support to TensorRT (#6150) 
* Add support to TensorRT

* Removed License header

* Bind input/output by position

* Comments

* More comments

* Add benchmark

* Add warning for performance degradation on large batch

* Address comments

* comments
 2018-04-11 17:03:54 -07:00