* Fixed issue #3069 by adding in_channels * Registerd group_conv2d_nchw as topi compute * Improved by checking tag value * Removed group_conv2d_nchw topi registration * Added test for relay group_conv2d_nchw * Added assertions to forbid small group size * Removed hard-coded oc_block_factor * Added explanatory comments to group_conv2d_nchw_cuda * Updated group_conv2d_nchw_cuda schedule Removed 'direct' CUDA tests * Reverted an accidental change in a conv2d test * Fixed indentation problems * Fixed a mis-commented line * Reverted change in group_conv2d_nchw tag * Removed commented int8 group_conv2d test * Fixed group size assertions in group_conv2d_nchw_cuda
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
rcnn | Loading commit data... | |
ssd | Loading commit data... | |
__init__.py | Loading commit data... | |
batch_matmul.py | Loading commit data... | |
conv2d.py | Loading commit data... | |
conv2d_direct.py | Loading commit data... | |
conv2d_hwcn.py | Loading commit data... | |
conv2d_int8.py | Loading commit data... | |
conv2d_transpose_nchw.py | Loading commit data... | |
conv2d_winograd.py | Loading commit data... | |
deformable_conv2d.py | Loading commit data... | |
dense.py | Loading commit data... | |
depthwise_conv2d.py | Loading commit data... | |
extern.py | Loading commit data... | |
group_conv2d_nchw.py | Loading commit data... | |
injective.py | Loading commit data... | |
nms.py | Loading commit data... | |
nn.py | Loading commit data... | |
pooling.py | Loading commit data... | |
reduction.py | Loading commit data... | |
softmax.py | Loading commit data... | |
tensor_intrin.py | Loading commit data... | |
vision.py | Loading commit data... |