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.

handle.cpp 2.9 kB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. #include "src/fallback/handle.h"
  2. #include "src/common/handle_impl.h"
  3. #include "src/fallback/add_update/opr_impl.h"
  4. #include "src/fallback/batched_matrix_mul/opr_impl.h"
  5. #include "src/fallback/concat/opr_impl.h"
  6. #include "src/fallback/conv_bias/opr_impl.h"
  7. #include "src/fallback/convolution/opr_impl.h"
  8. #include "src/fallback/elemwise/opr_impl.h"
  9. #include "src/fallback/elemwise_multi_type/opr_impl.h"
  10. #include "src/fallback/flip/opr_impl.h"
  11. #include "src/fallback/gaussian_blur/opr_impl.h"
  12. #include "src/fallback/group_local/opr_impl.h"
  13. #include "src/fallback/mask_conv/opr_impl.h"
  14. #include "src/fallback/matrix_mul/opr_impl.h"
  15. #include "src/fallback/pooling/opr_impl.h"
  16. #include "src/fallback/powc/opr_impl.h"
  17. #include "src/fallback/reduce/opr_impl.h"
  18. #include "src/fallback/relayout/opr_impl.h"
  19. #include "src/fallback/repeat/opr_impl.h"
  20. #include "src/fallback/resize/opr_impl.h"
  21. #include "src/fallback/roi_copy/opr_impl.h"
  22. #include "src/fallback/rotate/opr_impl.h"
  23. #include "src/fallback/softmax/opr_impl.h"
  24. #include "src/fallback/split/opr_impl.h"
  25. #include "src/fallback/tile/opr_impl.h"
  26. #include "src/fallback/type_cvt/opr_impl.h"
  27. #include "src/fallback/warp_perspective/opr_impl.h"
  28. namespace megdnn {
  29. namespace fallback {
  30. template <typename Opr>
  31. std::unique_ptr<Opr> HandleImpl::create_operator() {
  32. return naive::HandleImpl::create_operator<Opr>();
  33. }
  34. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Convolution)
  35. MEGDNN_SPECIALIZE_CREATE_OPERATOR(ConvolutionBackwardData)
  36. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Elemwise)
  37. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Pooling)
  38. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Reduce)
  39. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Concat)
  40. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Split)
  41. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Tile)
  42. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Repeat)
  43. MEGDNN_SPECIALIZE_CREATE_OPERATOR(RelayoutForward)
  44. MEGDNN_SPECIALIZE_CREATE_OPERATOR(MatrixMul)
  45. MEGDNN_SPECIALIZE_CREATE_OPERATOR(WarpPerspective)
  46. MEGDNN_SPECIALIZE_CREATE_OPERATOR(TypeCvt)
  47. MEGDNN_SPECIALIZE_CREATE_OPERATOR(GroupLocal)
  48. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Flip)
  49. MEGDNN_SPECIALIZE_CREATE_OPERATOR(GaussianBlur)
  50. MEGDNN_SPECIALIZE_CREATE_OPERATOR(SoftmaxForward)
  51. MEGDNN_SPECIALIZE_CREATE_OPERATOR(ROICopy)
  52. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Rotate)
  53. MEGDNN_SPECIALIZE_CREATE_OPERATOR(ElemwiseMultiType)
  54. MEGDNN_SPECIALIZE_CREATE_OPERATOR(AddUpdate)
  55. MEGDNN_SPECIALIZE_CREATE_OPERATOR(MaskConvForward)
  56. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Resize)
  57. MEGDNN_SPECIALIZE_CREATE_OPERATOR(BatchedMatrixMulForward)
  58. MEGDNN_SPECIALIZE_CREATE_OPERATOR(ConvBias)
  59. MEGDNN_SPECIALIZE_CREATE_OPERATOR(PowC)
  60. #pragma GCC diagnostic push
  61. #pragma GCC diagnostic ignored "-Wpragmas"
  62. #pragma GCC diagnostic ignored "-Winstantiation-after-specialization"
  63. MEGDNN_FOREACH_OPR_CLASS(MEGDNN_INST_CREATE_OPERATOR)
  64. #pragma GCC diagnostic pop
  65. } // namespace fallback
  66. } // namespace megdnn
  67. // vim: syntax=cpp.doxygen