tkDNN icon indicating copy to clipboard operation
tkDNN copied to clipboard

NetworkRT doesn't support 2D input and output layers

Open nightduck opened this issue 5 years ago • 1 comments

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.

nightduck avatar Dec 15 '20 22:12 nightduck

Thank you, we'll look into that in the next days

mive93 avatar Jan 04 '21 15:01 mive93