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

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!