Commit Graph

11 Commits (d84eb9b33f8751e50c73c07ed0d88379d9a406e9)

Author SHA1 Message Date
Zhaolong Xing 61238d31f7
Trt fp16 support (#18860)
6 years ago
nhzlx ec213730bc fix trt stream bug.
6 years ago
nhzlx 1600ba86f6 1. change tensorrt op from cpu to gpu
7 years ago
nhzlx 0015df1b12 modify op converter for conv2d
7 years ago
Yan Chunwei 4f95bc9463
feature/trt engine op test (#11182)
7 years ago
Yan Chunwei 0c0c5df4cb
feature/add TRT fc converter (#11043)
7 years ago
Yan Chunwei f5fc9c3bc1
feature/mul converter (#10841)
7 years ago
Yan Chunwei 674bd839cd
OpConverter change BlockDesc to proto::BlockDesc (#10623)
7 years ago
Luo Tao 53b401d589 refine io_convert and op_convert
7 years ago
Luo Tao beb1245560 add relu converter and unit-test
7 years ago
Luo Tao 6f6f330423 update the register method
7 years ago