equinox icon indicating copy to clipboard operation
equinox copied to clipboard

Support for data format NHWC in convolutional layers

Open MathiesW opened this issue 11 months ago • 5 comments

Hello, First of all, thank you for your great work! I noticed that Equinoxs' convolutional layers use NCHW data format, which, according to Nvidia Tensor Layout, is not optimal for Tensor cores on modern day GPUs. Changing the (default) data format from NCHW to NHWC hence may benefit performance of convolutional operations.

MathiesW avatar Jul 28 '23 14:07 MathiesW