Commit Graph

20 Commits (b4c826c5483e880afbdc9e96f0e793e085fffbb0)

Author SHA1 Message Date
Yiqun Liu 8bc1c5d2ab
Implement the Tensorrt plugin for elementwise op (#14487)
7 years ago
nhzlx c6a5c4b0c0 add comments for execute in ut_helper
7 years ago
nhzlx 1bf9d9e90c fix comments
7 years ago
nhzlx 144b20c160 add batch norm op converter
7 years ago
nhzlx 1600ba86f6 1. change tensorrt op from cpu to gpu
7 years ago
nhzlx 641f32da8c add softmax op converter
7 years ago
nhzlx 64a08f840f increase the test batch
7 years ago
nhzlx 4f71a3b12b fix a bug
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 bcd67bdd71 add assert for GetOutput
7 years ago
nhzlx 82527696e7 1. we delelte mul op, 2.modify fc and action op 3. modify the test inferface
7 years ago
gongweibao d9de6b8621
Add brpc surpport. (#11263)
7 years ago
Yan Chunwei 145aaa4b49
loose threshold of TRT for CI in different model (#11305)
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
fengjiayi d6997e5bc8
Merge pull request #11083 from JiayiFeng/dev_refine_programdesc_copy
7 years ago
fengjiayi 04ccbed5b8 fix a compile error
7 years ago
gongweibao 4fb7cc7f5e
Move sync_mode device ctx from grpc server (#10881)
7 years ago
Yan Chunwei f5fc9c3bc1
feature/mul converter (#10841)
7 years ago