I’m using TensorRT 2.1 and integrated customize layers by using all the plugin factory features (without the caffe parser).
I’m actually working on a network topology wich use a plugin layer that change the batch size between the input tensor and the output tensor.
Example with “LayerType” to describe topology
Actually it seems that the only way to do that in TensorRT is to return 32 in the “virtual int getNbOutputs()” method and to call 32 different time [Convolution]i[/i] after MyPlugin layer by pluged it to the different 32 MyPlugin output layers.
However it seems to be inefficient and i would like if there is another method to do that in TensorRT 2.1?