Hi,
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 ?
Thanks.