cudnnPoolingForward() tensor format support

Robotics & Edge Computing Jetson & Embedded Systems Jetson TK1
1

cuDNN 2.0 on CUDA 6.5 (Jetson TK1 / L4T)

This function only appears to support the NCHW tensor format, but not the NHWC format. Clearly the NCHW format is going to offer speed advantages, but is NHWC completely unsupported?

2

Hi,

It’s recommended to use TX1 or TX2 for deep learning use case.

NHWC is supported in our newer cuDNN version but is not available for TK1.
[url]Developer Guide :: NVIDIA Deep Learning cuDNN Documentation

Thanks.