cudnn-frontend icon indicating copy to clipboard operation
cudnn-frontend copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA

question about memory layout of the convolution

Open jinz2014 opened this issue 8 months ago • 1 comments

How could I disable channel-last memory layout in the graph.conv_fprop(...) function ? So the layout is NCHW instead. Thanks.

jinz2014 avatar Jun 14 '24 19:06 jinz2014