_nn.py
23 KB
-
[Topi] Tensorcore support for Conv3D (#5284) · 3df8d560
* one weird trick. * Added schedule knob for different workloads. * Initial conv3d tensorcore working. * Added conv3d tensorcore strategy. * Added layout conversion to tensorcore friendly format for conv2d and conv3d. * Add target name check. * Fixed bad names and depthwise check. * Removed duplicated attribute assignment.
Josh Fromm committed