Google Git
Sign in
apache/mxnet/refs/heads/leezu-patch-4/./src/operator/contrib
tree: eb87015250d80c5d6b03f71eac8134943af1c2c7
  1. intgemm/
  2. nn/
  3. tvmop/
  4. adamw-inl.h
  5. adamw.cc
  6. adamw.cu
  7. adaptive_avg_pooling-inl.h
  8. adaptive_avg_pooling.cc
  9. adaptive_avg_pooling.cu
  10. all_finite-inl.h
  11. all_finite.cc
  12. all_finite.cu
  13. allclose_op-inl.h
  14. allclose_op.cc
  15. allclose_op.cu
  16. amp_graph_pass.cc
  17. batch_norm_relu.cc
  18. bilinear_resize-inl.cuh
  19. bilinear_resize-inl.h
  20. bilinear_resize.cc
  21. bilinear_resize.cu
  22. boolean_mask-inl.h
  23. boolean_mask.cc
  24. boolean_mask.cu
  25. bounding_box-common.h
  26. bounding_box-inl.cuh
  27. bounding_box-inl.h
  28. bounding_box.cc
  29. bounding_box.cu
  30. count_sketch-inl.h
  31. count_sketch.cc
  32. count_sketch.cu
  33. deformable_psroi_pooling-inl.h
  34. deformable_psroi_pooling.cc
  35. deformable_psroi_pooling.cu
  36. dgl_graph-inl.h
  37. dgl_graph.cc
  38. dgl_graph.cu
  39. dynamic_shape_ops-inl.h
  40. dynamic_shape_ops.cc
  41. erfinv-inl.h
  42. fft-inl.h
  43. fft.cc
  44. fft.cu
  45. gradient_multiplier_op.cc
  46. gradient_multiplier_op.cu
  47. hawkes_ll-inl.h
  48. hawkes_ll.cc
  49. hawkes_ll.cu
  50. index_array-inl.h
  51. index_array.cc
  52. index_array.cu
  53. index_copy-inl.h
  54. index_copy.cc
  55. index_copy.cu
  56. krprod.cc
  57. krprod.h
  58. mrcnn_mask_target-inl.h
  59. mrcnn_mask_target.cu
  60. multi_lamb-inl.h
  61. multi_lamb.cc
  62. multi_lamb.cu
  63. multi_lans-inl.h
  64. multi_lans.cc
  65. multi_lans.cu
  66. multi_lars-inl.h
  67. multi_lars.cc
  68. multi_lars.cu
  69. multi_proposal-inl.h
  70. multi_proposal.cc
  71. multi_proposal.cu
  72. multi_sum_sq-inl.h
  73. multi_sum_sq.cc
  74. multi_sum_sq.cu
  75. multibox_detection-inl.h
  76. multibox_detection.cc
  77. multibox_detection.cu
  78. multibox_prior-inl.h
  79. multibox_prior.cc
  80. multibox_prior.cu
  81. multibox_target-inl.h
  82. multibox_target.cc
  83. multibox_target.cu
  84. nnz.cc
  85. optimizer_op-inl.h
  86. optimizer_op.cc
  87. optimizer_op.cu
  88. preloaded_multi_sgd-inl.h
  89. preloaded_multi_sgd.cc
  90. preloaded_multi_sgd.cu
  91. proposal-inl.h
  92. proposal.cc
  93. proposal.cu
  94. psroi_pooling-inl.h
  95. psroi_pooling.cc
  96. psroi_pooling.cu
  97. quadratic_op-inl.h
  98. quadratic_op.cc
  99. quadratic_op.cu
  100. reset_arrays-inl.h
  101. reset_arrays.cc
  102. reset_arrays.cu
  103. roi_align-inl.h
  104. roi_align.cc
  105. roi_align.cu
  106. rroi_align-inl.h
  107. rroi_align.cc
  108. stes_op.cc
  109. stes_op.cu
  110. stes_op.h
  111. sync_batch_norm-inl.h
  112. sync_batch_norm.cc
  113. sync_batch_norm.cu
  114. transformer-inl.h
  115. transformer.cc
  116. transformer.cu