Hello,
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
[Convolution]i-[/i]>[Convolution]i[/i]->[MyPlugin]i[/i]->[Convolution]i[/i]
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?
Thanks