topi/python/topi/cuda · ff6fa3995c6b2cd253c6265798567a635dc00aa8 · wenyuanbo / tic

[TOPI][Tensor Core] Conv2d and Dense ops support on Tensor Core (#5099) · f4286cc7

* [TOPI][Tensor Core] Optimization of CNNs on Tensor Core #6004

* update conv2d test

* # pylint: dense_tensorcore.py

* modify

* modify conv2d

* modify the unclear comment,add shape assertion in conv2d compute,combine general gemm intrinsic

* add shape assertion in conv2d compute, combine general gemm intrinsic

Co-authored-by: libaihong <libaihong@inspur.com>
Co-authored-by: libaihong <61525430+libaihong@users.noreply.github.com>

committed Mar 27, 2020

f4286cc7

Name	Last commit	Last update
..
rcnn		Loading commit data...
ssd		Loading commit data...
__init__.py		Loading commit data...
batch_matmul.py		Loading commit data...
conv1d.py		Loading commit data...
conv1d_transpose_ncw.py		Loading commit data...
conv2d.py		Loading commit data...
conv2d_alter_op.py		Loading commit data...
conv2d_direct.py		Loading commit data...
conv2d_hwcn.py		Loading commit data...
conv2d_int8.py		Loading commit data...
conv2d_nhwc.py		Loading commit data...
conv2d_nhwc_tensorcore.py		Loading commit data...
conv2d_transpose_nchw.py		Loading commit data...
conv2d_winograd.py		Loading commit data...
conv3d.py		Loading commit data...
conv3d_direct.py		Loading commit data...
deformable_conv2d.py		Loading commit data...
dense.py		Loading commit data...
dense_tensorcore.py		Loading commit data...
depthwise_conv2d.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...
sort.py		Loading commit data...
tensor_intrin.py		Loading commit data...
vision.py		Loading commit data...