Hello. Does tensorrt4 support 3D convolution layer and 3D pooling layer of Keras? If not, how can I add these layers to TensorRT4 in python? Thanks. Waiting for reply
Assuming you are using Keras with Tensorflow backend, please reference https://docs.nvidia.com/deeplearning/sdk/tensorrt-support-matrix/index.html#layers-matrix for a list of supported layers.
For unsupported layers, users can extend TensorRT functionalities by implementing custom layers using the IPluginV2 class for the C++ and Python API. Custom layers, often referred to as plugins, are implemented and instantiated by an application, and their lifetime must span their use within a TensorRT engine. https://docs.nvidia.com/deeplearning/sdk/tensorrt-developer-guide/index.html#extending
NVIDIA Enterprise Support