onnx.py
63 KB
-
Conv3D ONNX support and conv3D_ncdhw x86 schedules (#4949) · d56829ea
* Support 3d Convolution with the ONNX frontend * add unit tests for conv3d in onnx frontend respond to PR formatting requests add x86 schedules to conv3d ncdhw test fix a doc string format issue refactor for changed upsream API * first attempt at conv3d autotuning add default schedule for conv3d_ncdhw fill in autotvm integration add a fallback for invalid schedules fix fallback fix reduction order to get simd working correctly
Matthew Brookhart committed