Tristan Konolige 2c0a7c2a7e [TOPI] Add support for groupped conv3d (#9873) ...

* [TOPI] Add support for groupped conv3d

Change conv3d to use generic conv implementation which supports groupped
convolutions. Also, remove support for non-float16 tensorcore operations
as they cause large degradation in accuracy. Generic conv now supports
autoscheduler.

* correct none check

* add tests for floordiv simplification

* fixed incorrect test for autoscheduler

* formatting

* add groups to winograd

* fix tensorcore

* manually simplify index instead of relying on simplifier

* formatting

* add groups argument to conv3d_ncdhw_winograd_without_weight_transform

* formatting

2022-02-19 06:18:01 +09:00

..

__init__.py

[CUDNN] Support gradient kernels (#9986)

2022-01-23 06:58:31 +09:00

adaptive_pool_python.py

fix py files (#8194)

2021-06-04 23:22:46 +01:00

batch_matmul.py

[TOPI] Add transpose_a/b & dynamic shape support for batch matmul (#8527)

2021-07-29 10:15:21 -07:00

batch_norm.py

[TOPI] Add generic batch norm (#9694)

2021-12-13 09:00:22 -07:00

batch_to_space_nd.py

[Relay] Add space_to_batch_nd and batch_to_space_nd operators (#6477)

2020-11-17 21:04:12 +08:00

common.py

[Topi][Testing] Float16 unittests for dense, conv2d, depthwise conv2d (#8529)

2021-08-07 11:26:22 +09:00

conv1d_ncw_python.py

[TOPI] Support grouped conv1d (#9832)

2022-01-07 16:27:42 -07:00

conv1d_transpose_ncw_python.py

[REFACTOR] Remainings of util => utils (#6778)

2020-10-29 13:46:31 -04:00

conv2d_backcward_weight_python.py

[CUTLASS] Initial support for conv2d wgrad (#10177)

2022-02-08 10:28:20 +09:00

conv2d_hwcn_python.py

[BugFix]: Convert tuple to int (#7880)

2021-04-25 15:21:37 +08:00

conv2d_nchw_python.py

[Topi][Testing] Float16 unittests for dense, conv2d, depthwise conv2d (#8529)

2021-08-07 11:26:22 +09:00

conv2d_nhwc_python.py

[REFACTOR] Remainings of util => utils (#6778)

2020-10-29 13:46:31 -04:00

conv2d_transpose_python.py

[CUDNN] Support gradient kernels (#9986)

2022-01-23 06:58:31 +09:00

conv3d_ncdhw_python.py

[REFACTOR] Remainings of util => utils (#6778)

2020-10-29 13:46:31 -04:00

conv3d_ndhwc_python.py

[TOPI] Add support for groupped conv3d (#9873)

2022-02-19 06:18:01 +09:00

conv3d_transpose_ncdhw_python.py

[REFACTOR] Remainings of util => utils (#6778)

2020-10-29 13:46:31 -04:00

correlation_nchw_python.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

crop_and_resize_python.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

deformable_conv2d_python.py

Fix Bug in Bilinear Interpolation and Add Deform Conv to PT FrontEnd (#7397)

2021-02-04 20:49:07 -08:00

dense.py

[CUDA] dense_tensorcore/batch_matmul_tensorcore support int8/int4 (#8402)

2021-07-09 17:40:40 +08:00

depth_to_space.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

depthwise_conv2d_python.py

[Topi][Testing] Float16 unittests for dense, conv2d, depthwise conv2d (#8529)

2021-08-07 11:26:22 +09:00

dilate_python.py

[Topi][Testing] Float16 unittests for dense, conv2d, depthwise conv2d (#8529)

2021-08-07 11:26:22 +09:00

gather_nd_python.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

gather_python.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

grid_sample_python.py

Support PyTorch grid_sample (#10184)

2022-02-08 14:57:15 +09:00

l2_normalize_python.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

lrn_python.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

matrix_set_diag.py

[Relay/TOPI] Added 'offsets' and 'alignment' attributes to MATRIX_SET_DIAG. (#6429)

2020-09-24 08:43:28 -07:00

nll_loss.py

[Relay, TOPI] Add negative log likelihood loss (nll_loss) op (#8056)

2021-06-25 11:46:48 -04:00

one_hot.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

pool_grad_python.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

poolnd_python.py

[TOPI] Fix nn.pool*d issue with 'vectorize' function and add unit tests (#8541)

2021-07-29 19:53:02 +09:00

reorg_python.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

resize_python.py

speed up reference resize kernel (#8592)

2021-07-30 09:49:19 -04:00

roi_align_python.py

[TOPI, Relay] Support roi_align NHWC layout (#7463)

2021-02-18 12:11:17 +09:00

roi_pool_python.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

searchsorted.py

[Relay, TOPI] Add searchsorted op (#9184)

2021-10-20 18:19:36 -04:00

sequence_mask_python.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

slice_axis_python.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

softmax_python.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

space_to_batch_nd.py

[Relay] Add space_to_batch_nd and batch_to_space_nd operators (#6477)

2020-11-17 21:04:12 +08:00

space_to_depth.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

strided_slice_python.py

[Relay, TOPI] Refactor strided_slice and add axes argument (#8165)

2021-06-03 06:33:00 +09:00