* [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>
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
__init__.py | Loading commit data... | |
config.py | Loading commit data... | |
darknet.py | Loading commit data... | |
dcgan.py | Loading commit data... | |
densenet.py | Loading commit data... | |
dqn.py | Loading commit data... | |
inception_v3.py | Loading commit data... | |
init.py | Loading commit data... | |
layers.py | Loading commit data... | |
lstm.py | Loading commit data... | |
mlp.py | Loading commit data... | |
mobilenet.py | Loading commit data... | |
nat.py | Loading commit data... | |
py_converter.py | Loading commit data... | |
resnet.py | Loading commit data... | |
squeezenet.py | Loading commit data... | |
temp_op_attr.py | Loading commit data... | |
tf.py | Loading commit data... | |
vgg.py | Loading commit data... | |
yolo_detection.py | Loading commit data... |