i am working on Space to Batch and Batch to Space plugin because in my model there are some Conv1d with dilation rate > 1 and TensorFlow will convert this layer into some operations as below
Conv1d with dilation rate > 1
is converted to
SpaceToBatch: reduce spatial size and increase batch size according to dilation rate (batchSize_1 is changed to batchSize_2) ExpandsDims : insert one dimension for the following Conv2D (batchSize_2) Conv2d : performance normal 2d convolutional operation (batchSize_2) Squeeze : remove inserted dimension (batchSize_2) BatchToSpace: reverse of SpaceToBatch (batchSize_1)
my question is either SpaceToBatch or BatchToSpace will change batch size but it seems i can not change batch size in the plugin ?
- For getOutputDimension(int index, const nvinfer1:Dims *inputs, int nbInputDims), can i change batch size here ?