Commit Graph

14 Commits (d84eb9b33f8751e50c73c07ed0d88379d9a406e9)

Author SHA1 Message Date
Zhaolong Xing 26ae6d49e4
Update trt5 for paddle-trt (#18645)
6 years ago
nhzlx d40402f9b7 add dropout and sigmoid op converter
7 years ago
nhzlx 5533400720 fix comments
7 years ago
nhzlx 01566fb61b 1. support mutil batch utest 2. support pool op
7 years ago
nhzlx 7382f98600 1. set ut batch > 1 2. readd the mul op(utest will be added later)
7 years ago
nhzlx 82527696e7 1. we delelte mul op, 2.modify fc and action op 3. modify the test inferface
7 years ago
Luo Tao e116129f03 rewrite unittest of trt_activation_op
7 years ago
Yan Chunwei f5fc9c3bc1
feature/mul converter (#10841)
7 years ago
Luo Tao 1992f70920 Merge branch 'develop' into refine_relu_test
7 years ago
Yan Chunwei 674bd839cd
OpConverter change BlockDesc to proto::BlockDesc (#10623)
7 years ago
Luo Tao 4f5f0be769 use the latest buffer to update the convert
7 years ago
Luo Tao 89dcb0bd15 refine EngineIOConverter, and use io_convert in test_trt_activation_op
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