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