**Is your feature request related to a problem? Please describe.** <!--A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]--> The current implementation uses a validator to handle the accumulate=True case. https://github.com/pytorch/TensorRT/pull/2880 **Describe the solution you'd like** <!--A clear and concise description of what you want to happen.--> **Describe alternatives you've considered** <!--A clear and concise description of any alternative solutions or features you've considered.--> **Additional context** <!--Add any other context or screenshots about the feature request here.-->
