Google Git
Sign in
apache / tvm / fc814e704138bbb0d24cee7c77919e9bf3e01d7d / . / python / tvm / topi / cuda
tree: d45924424101f2d239c743a3a946f16afa25cc17 [path history] [tgz]
  1. rcnn/
  2. ssd/
  3. __init__.py
  4. argwhere.py
  5. batch_matmul.py
  6. batch_matmul_tensorcore.py
  7. conv1d.py
  8. conv1d_transpose_ncw.py
  9. conv2d.py
  10. conv2d_alter_op.py
  11. conv2d_direct.py
  12. conv2d_hwcn.py
  13. conv2d_hwnc_tensorcore.py
  14. conv2d_int8.py
  15. conv2d_nhwc_tensorcore.py
  16. conv2d_nhwc_winograd.py
  17. conv2d_transpose.py
  18. conv2d_winograd.py
  19. conv3d.py
  20. conv3d_alter_op.py
  21. conv3d_direct.py
  22. conv3d_ndhwc_tensorcore.py
  23. conv3d_transpose_ncdhw.py
  24. conv3d_winograd.py
  25. correlation.py
  26. deformable_conv2d.py
  27. dense.py
  28. dense_tensorcore.py
  29. depthwise_conv2d.py
  30. group_conv2d_nchw.py
  31. injective.py
  32. nms.py
  33. nn.py
  34. pooling.py
  35. reduction.py
  36. scan.py
  37. scatter.py
  38. scatter_elements.py
  39. searchsorted.py
  40. signal.py
  41. softmax.py
  42. sort.py
  43. sparse.py
  44. sparse_reshape.py
  45. tensor_intrin.py
  46. tensorcore_alter_op.py
  47. transform.py
  48. unique.py
  49. vision.py