Optimize TensorRT Operator

StarCraft2 Model

Operator

Shape

TensorRT-7.2.2.3

TPAT Plugin

Improvement

Resize

NCHW(32, 32, 44, 50) ->

(32, 32, 176, 200)

1.450 ms

1.079 ms

1.38x

NCHW(32, 64, 11, 12) ->

(32, 64, 176, 200)

2.759 ms

2.171 ms

1.27x

Reduce_max

input_shape=(256, 82, 128)
output_shape=(256, 128)

0.167 ms

0.036 ms

4.63x