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