Google Git
Sign in
apache/mxnet/refs/heads/benchmark/./src/operator/contrib
tree: 62eb75ab6ebaefefa59aca1b034d6380d29796b5
  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. amp_graph_pass.cc
  13. bilinear_resize-inl.cuh
  14. bilinear_resize-inl.h
  15. bilinear_resize.cc
  16. bilinear_resize.cu
  17. boolean_mask-inl.h
  18. boolean_mask.cc
  19. boolean_mask.cu
  20. bounding_box-common.h
  21. bounding_box-inl.cuh
  22. bounding_box-inl.h
  23. bounding_box.cc
  24. bounding_box.cu
  25. count_sketch-inl.h
  26. count_sketch.cc
  27. count_sketch.cu
  28. deformable_convolution-inl.h
  29. deformable_convolution.cc
  30. deformable_convolution.cu
  31. deformable_psroi_pooling-inl.h
  32. deformable_psroi_pooling.cc
  33. deformable_psroi_pooling.cu
  34. dgl_graph-inl.h
  35. dgl_graph.cc
  36. dgl_graph.cu
  37. erfinv-inl.h
  38. fft-inl.h
  39. fft.cc
  40. fft.cu
  41. gradient_multiplier_op.cc
  42. gradient_multiplier_op.cu
  43. hawkes_ll-inl.h
  44. hawkes_ll.cc
  45. hawkes_ll.cu
  46. ifft-inl.h
  47. ifft.cc
  48. ifft.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_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. sync_batch_norm-inl.h
  105. sync_batch_norm.cc
  106. sync_batch_norm.cu
  107. transformer-inl.h
  108. transformer.cc
  109. transformer.cu