tkDNN
tkDNN copied to clipboard
NetworkRT doesn't support 2D input and output layers
The layers used in the dataDim_t struct is naturally 5 dimensional. NetworkRT reduces them to 3 by filling dimensions in with a 1. Eg 1x224x224x3x1. However, calling getBindingDimensions on a layer will return a struct that pads it with zeros, due to a quirk of a TensorRT call. dataDim_t has a method that will return the total number of elements in it, but if a dimension is zero, this total is always zero. So all 2D layers look empty.
I'm submitted a pull request so the dataDim_t struct can be initialized in a way to correct these zeros into ones.
Thank you, we'll look into that in the next days