YuzhouPeng / tensorRT_Wheels

tensorrt wheels (API)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

tensorRT_Wheels

tensorrt wheels (API)

#Current wheels(API):

1.input 2.conv 3.pool 4.batchNorm 5.activation(relu/sigmoid/tanh/leaky_relu/elu/selu/softsign/softplus/clip/hard_sigmoid/scaled_tanh/thresholded_relu) 6.FC( == linear) 7.elementWise(sum/sub/prod/div/power...) 8.concat 9.slice 10.softmax 11.matmul 12.shuffle_reshape( just reshape) 13.shuffle_permute( just permute) 14.shuffle ( reshape and permute) 15.reduce 16.topk (argmax/argmin) 17.upsample (nerest/bilinear) 18.constant ("The most important layer!!!") 19.PReLU 20.groupNorm <<<<<<< HEAD 21.padding

02e7460caefd01e997b4d5d7ca45657173cf3025

About

tensorrt wheels (API)


Languages

Language:C++ 100.0%