Google Git
Sign in
apache/mxnet/refs/heads/v1.x/./src/operator/contrib
tree: d6a347592c9396260cdfa4d4d9622f5e2ac7ca3d [path history] [tgz]
  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_convolution-inl.h
  34. deformable_convolution.cc
  35. deformable_convolution.cu
  36. deformable_psroi_pooling-inl.h
  37. deformable_psroi_pooling.cc
  38. deformable_psroi_pooling.cu
  39. dgl_graph-inl.h
  40. dgl_graph.cc
  41. dgl_graph.cu
  42. erfinv-inl.h
  43. fft-inl.h
  44. fft.cc
  45. fft.cu
  46. gradient_multiplier_op.cc
  47. gradient_multiplier_op.cu
  48. hawkes_ll-inl.h
  49. hawkes_ll.cc
  50. hawkes_ll.cu
  51. ifft-inl.h
  52. ifft.cc
  53. ifft.cu
  54. index_array-inl.h
  55. index_array.cc
  56. index_array.cu
  57. index_copy-inl.h
  58. index_copy.cc
  59. index_copy.cu
  60. krprod.cc
  61. krprod.h
  62. modulated_deformable_convolution-inl.h
  63. modulated_deformable_convolution.cc
  64. modulated_deformable_convolution.cu
  65. mrcnn_mask_target-inl.h
  66. mrcnn_mask_target.cu
  67. multi_lamb-inl.h
  68. multi_lamb.cc
  69. multi_lamb.cu
  70. multi_lars-inl.h
  71. multi_lars.cc
  72. multi_lars.cu
  73. multi_proposal-inl.h
  74. multi_proposal.cc
  75. multi_proposal.cu
  76. multi_sum_sq-inl.h
  77. multi_sum_sq.cc
  78. multi_sum_sq.cu
  79. multibox_detection-inl.h
  80. multibox_detection.cc
  81. multibox_detection.cu
  82. multibox_prior-inl.h
  83. multibox_prior.cc
  84. multibox_prior.cu
  85. multibox_target-inl.h
  86. multibox_target.cc
  87. multibox_target.cu
  88. nnz.cc
  89. optimizer_op-inl.h
  90. optimizer_op.cc
  91. optimizer_op.cu
  92. preloaded_multi_sgd-inl.h
  93. preloaded_multi_sgd.cc
  94. preloaded_multi_sgd.cu
  95. proposal-inl.h
  96. proposal.cc
  97. proposal.cu
  98. psroi_pooling-inl.h
  99. psroi_pooling.cc
  100. psroi_pooling.cu
  101. quadratic_op-inl.h
  102. quadratic_op.cc
  103. quadratic_op.cu
  104. reset_arrays-inl.h
  105. reset_arrays.cc
  106. reset_arrays.cu
  107. roi_align-inl.h
  108. roi_align.cc
  109. roi_align.cu
  110. rroi_align-inl.h
  111. rroi_align.cc
  112. stes_op.cc
  113. stes_op.cu
  114. stes_op.h
  115. sync_batch_norm-inl.h
  116. sync_batch_norm.cc
  117. sync_batch_norm.cu
  118. transformer-inl.h
  119. transformer.cc
  120. transformer.cu