Skip to content

Output Dimension

Xinyu Wang edited this page Jun 13, 2024 · 1 revision

NVDLA pads it output to nearest multiple of 64(or 32).

For general usage, output has to be reformatted, see reformat function for details.

For example, if output is 1 * 3 * 60 * 60 of __half format, then the actual output dimension of cuDLA would be 1 * 3 * 60 * 64

Clone this wiki locally