Closed milhidaka closed 2 years ago
im2col step in Conv produces large temporary tensor, which sometimes exceeds texture size limit. The new code automatically divides tensor to avoid large texture.
im2col step in Conv produces large temporary tensor, which sometimes exceeds texture size limit. The new code automatically divides tensor to avoid large texture.