OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: b3f46734e7
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'b3f46734e7'
${ noResults }
MegEngine/dnn/src/fallback/handle.cpp

75 lines
2.9 kB
Raw Blame History 

  1. #include "src/fallback/handle.h"

  2. #include "src/common/handle_impl.h"

  3. 

  4. #include "src/fallback/add_update/opr_impl.h"

  5. #include "src/fallback/batched_matrix_mul/opr_impl.h"

  6. #include "src/fallback/concat/opr_impl.h"

  7. #include "src/fallback/conv_bias/opr_impl.h"

  8. #include "src/fallback/convolution/opr_impl.h"

  9. #include "src/fallback/elemwise/opr_impl.h"

  10. #include "src/fallback/elemwise_multi_type/opr_impl.h"

  11. #include "src/fallback/flip/opr_impl.h"

  12. #include "src/fallback/gaussian_blur/opr_impl.h"

  13. #include "src/fallback/group_local/opr_impl.h"

  14. #include "src/fallback/mask_conv/opr_impl.h"

  15. #include "src/fallback/matrix_mul/opr_impl.h"

  16. #include "src/fallback/pooling/opr_impl.h"

  17. #include "src/fallback/powc/opr_impl.h"

  18. #include "src/fallback/reduce/opr_impl.h"

  19. #include "src/fallback/relayout/opr_impl.h"

  20. #include "src/fallback/repeat/opr_impl.h"

  21. #include "src/fallback/resize/opr_impl.h"

  22. #include "src/fallback/roi_copy/opr_impl.h"

  23. #include "src/fallback/rotate/opr_impl.h"

  24. #include "src/fallback/softmax/opr_impl.h"

  25. #include "src/fallback/split/opr_impl.h"

  26. #include "src/fallback/tile/opr_impl.h"

  27. #include "src/fallback/type_cvt/opr_impl.h"

  28. #include "src/fallback/warp_perspective/opr_impl.h"

  29. 

  30. namespace megdnn {

  31. namespace fallback {

  32. 

  33. template <typename Opr>

  34. std::unique_ptr<Opr> HandleImpl::create_operator() {

  35.     return naive::HandleImpl::create_operator<Opr>();

  36. }

  37. 

  38. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Convolution)

  39. MEGDNN_SPECIALIZE_CREATE_OPERATOR(ConvolutionBackwardData)

  40. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Elemwise)

  41. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Pooling)

  42. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Reduce)

  43. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Concat)

  44. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Split)

  45. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Tile)

  46. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Repeat)

  47. MEGDNN_SPECIALIZE_CREATE_OPERATOR(RelayoutForward)

  48. MEGDNN_SPECIALIZE_CREATE_OPERATOR(MatrixMul)

  49. MEGDNN_SPECIALIZE_CREATE_OPERATOR(WarpPerspective)

  50. MEGDNN_SPECIALIZE_CREATE_OPERATOR(TypeCvt)

  51. MEGDNN_SPECIALIZE_CREATE_OPERATOR(GroupLocal)

  52. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Flip)

  53. MEGDNN_SPECIALIZE_CREATE_OPERATOR(GaussianBlur)

  54. MEGDNN_SPECIALIZE_CREATE_OPERATOR(SoftmaxForward)

  55. MEGDNN_SPECIALIZE_CREATE_OPERATOR(ROICopy)

  56. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Rotate)

  57. MEGDNN_SPECIALIZE_CREATE_OPERATOR(ElemwiseMultiType)

  58. MEGDNN_SPECIALIZE_CREATE_OPERATOR(AddUpdate)

  59. MEGDNN_SPECIALIZE_CREATE_OPERATOR(MaskConvForward)

  60. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Resize)

  61. MEGDNN_SPECIALIZE_CREATE_OPERATOR(BatchedMatrixMulForward)

  62. MEGDNN_SPECIALIZE_CREATE_OPERATOR(ConvBias)

  63. MEGDNN_SPECIALIZE_CREATE_OPERATOR(PowC)

  64. 

  65. #pragma GCC diagnostic push

  66. #pragma GCC diagnostic ignored "-Wpragmas"

  67. #pragma GCC diagnostic ignored "-Winstantiation-after-specialization"

  68. MEGDNN_FOREACH_OPR_CLASS(MEGDNN_INST_CREATE_OPERATOR)

  69. #pragma GCC diagnostic pop

  70. 

  71. }  // namespace fallback

  72. }  // namespace megdnn

  73. 

  74. // vim: syntax=cpp.doxygen