conv_forward.cc
10.3 KB
-
[TOPI] add extern schedule for cudnn and miopen (#724) · cdb2f873* add extern schedule for miopen * fix comment * optionally dispatch to miopen from topi * fix lint * check if current target is None * use generic dispatch for rocm conv2d * fix lint * fix workspace bug * remove blank line * remove blank line * remove blank line
masahi committed