TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

PRelu is not integrated with QDQ

Open pangr opened this issue 3 years ago • 1 comments

Description

Environment

TensorRT Version 8.4: NVIDIA GPU 1080ti: NVIDIA Driver Version 470: CUDA Version 11.0: CUDNN Version: Operating System: Python Version (if applicable): Tensorflow Version (if applicable): PyTorch Version (if applicable): Baremetal or Container (if so, version):

Relevant Files

Steps To Reproduce

I customized the QDQ operation of PRelu, but the QDQ module of PRelu does not seem to be integrated with PRelu. The following is the model diagram and dumpprofile image image

pangr avatar Jul 18 '22 09:07 pangr

@ttyio ^ ^

zerollzeng avatar Jul 18 '22 16:07 zerollzeng

@pangr , sorry for the delay response. Could you remove the 2 DQ nodes before the plugin? If the PRelu output is also INT8, then also need remove the Q after the plugin.

ttyio avatar Dec 06 '22 01:12 ttyio

Closing since no activity for more than 3 weeks, please reopen if you still have question, thanks!

ttyio avatar Jan 10 '23 02:01 ttyio