How to specify 1 or 2-dimensional inputs in IUffParser::registerInput?

I’ve read the extremely sparse API documentation, looked through code samples, searched in the forums, but I’m not sure what’s the correct way for supposedly trivial tasks.


  1. How do I register an input shape that looks like [batch, time, channels] in the registerInput function, if UffInputOrder only has entries for kNCHW, KNHWC and KNC?

  2. Along the same vein, I suppose the right way to set an input with shape [batch, 1] would be something like this:

parser->registerInput("input", Dims{.nbDims=1, .d={1, ...}, .type={kCHANNEL, ...}}, UffInputOrder::KNC)

Is that assumption correct?