TensorRT

Description

Do we have any python API to add batch norm layer in Tensor RT .
How to add batch norm layer in Tensort RT using Python Tensort API python??

Environment

TensorRT Version: 8.0
GPU Type:
Nvidia Driver Version: 470.57.02
CUDA Version: 11.3
CUDNN Version: 8.2
Operating System + Version: ubuntu20.04
Python Version (if applicable): 3.8
TensorFlow Version (if applicable): 2.5
PyTorch Version (if applicable):
Baremetal or Container (if container which image + tag):

Relevant Files

Please attach or include links to any models, data, files, or scripts necessary to reproduce your issue. (Github repo, Google Drive, Dropbox, etc.)

Steps To Reproduce

Please include:

  • Exact steps/commands to build your repro
  • Exact steps/commands to run your repro
  • Full traceback of errors encountered

Hi,
Please check the below link, as they might answer your concerns
https://docs.nvidia.com/deeplearning/tensorrt/api/index.html
Thanks!

There is no Batch norm layer API in the given documents,Can you please refer any related documents.
Thanks!!

Hi @Ragu_23 ,
This post should help you with your query

Thanks!