How to disable NCHW to NHWC conversion of inputs and outputs?

The API doesn’t seem to support it and with Volta kernels natively NHWC this adds a lot of overhead for smaller graphs.

@kayccc can you please help with this? thank you