You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Yiqun Liu 8bc1c5d2ab Implement the Tensorrt plugin for elementwise op (#14487) ... * Initialize the elementwise plugin. * Implement the basic CUDA kernel of elementwise plugin. test=develop		6 years ago
..
CMakeLists.txt	Implement the Tensorrt plugin for elementwise op (#14487)	6 years ago
activation_op.cc	…
batch_norm_op.cc	…
concat_op.cc	…
conv2d_op.cc	Complete PRelu plugin and Conv2d transpose op converter	6 years ago
dropout_op.cc	…
elementwise_op.cc	Implement the Tensorrt plugin for elementwise op (#14487)	6 years ago
fc_op.cc	…
io_converter.cc	…
io_converter.h	…
mul_op.cc	…
op_converter.h	Implement the Tensorrt plugin for elementwise op (#14487)	6 years ago
pad_op.cc	…
pool2d_op.cc	…
prelu_op.cc	Implement the Tensorrt plugin for elementwise op (#14487)	6 years ago
softmax_op.cc	…
split_op.cc	Implement the Tensorrt plugin for elementwise op (#14487)	6 years ago
test_activation_op.cc	…
test_batch_norm_op.cc	…
test_concat_op.cc	…
test_conv2d_op.cc	Fix code style	6 years ago
test_dropout_op.cc	…
test_elementwise_op.cc	Implement the Tensorrt plugin for elementwise op (#14487)	6 years ago
test_fc_op.cc	…
test_io_converter.cc	…
test_mul_op.cc	Implement the Tensorrt plugin for elementwise op (#14487)	6 years ago
test_op_converter.cc	…
test_pad_op.cc	…
test_pool2d_op.cc	…
test_prelu_op.cc	Complete PRelu plugin and Conv2d transpose op converter	6 years ago
test_softmax_op.cc	…
test_split_op.cc	…
ut_helper.h	Implement the Tensorrt plugin for elementwise op (#14487)	6 years ago