wangwei
/
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: b8316de5a5
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'b8316de5a5'
${ noResults }
MegEngine/dnn/src/arm_common/handle.cpp

64 lines
2.3 kB
Raw Blame History 

  1. /**

  2.  * \file dnn/src/arm_common/handle.cpp

  3.  * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")

  4.  *

  5.  * Copyright (c) 2014-2021 Megvii Inc. All rights reserved.

  6.  *

  7.  * Unless required by applicable law or agreed to in writing,

  8.  * software distributed under the License is distributed on an

  9.  * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  10.  */

  11. 

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

  13. 

  14. #include "src/arm_common/handle.h"

  15. 

  16. #include "src/arm_common/conv_bias/opr_impl.h"

  17. #include "src/arm_common/convolution/opr_impl.h"

  18. #include "src/arm_common/cvt_color/opr_impl.h"

  19. #include "src/arm_common/elemwise/opr_impl.h"

  20. #include "src/arm_common/elemwise_multi_type/opr_impl.h"

  21. #include "src/arm_common/local/opr_impl.h"

  22. #include "src/arm_common/pooling/opr_impl.h"

  23. #include "src/arm_common/reduce/opr_impl.h"

  24. #include "src/arm_common/resize/opr_impl.h"

  25. #include "src/arm_common/separable_conv/opr_impl.h"

  26. #include "src/arm_common/separable_filter/opr_impl.h"

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

  28. #include "src/arm_common/warp_affine/opr_impl.h"

  29. #include "src/arm_common/warp_perspective/opr_impl.h"

  30. 

  31. namespace megdnn {

  32. namespace arm_common {

  33. 

  34. template <typename Opr>

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

  36.     return fallback::HandleImpl::create_operator<Opr>();

  37. }

  38. 

  39. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Pooling)

  40. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Local)

  41. MEGDNN_SPECIALIZE_CREATE_OPERATOR(SeparableConv)

  42. MEGDNN_SPECIALIZE_CREATE_OPERATOR(SeparableFilter)

  43. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Elemwise)

  44. MEGDNN_SPECIALIZE_CREATE_OPERATOR(ElemwiseMultiType)

  45. MEGDNN_SPECIALIZE_CREATE_OPERATOR(CvtColor)

  46. MEGDNN_SPECIALIZE_CREATE_OPERATOR(WarpAffine)

  47. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Resize)

  48. MEGDNN_SPECIALIZE_CREATE_OPERATOR(WarpPerspective)

  49. MEGDNN_SPECIALIZE_CREATE_OPERATOR(TypeCvt)

  50. MEGDNN_SPECIALIZE_CREATE_OPERATOR(Reduce)

  51. MEGDNN_SPECIALIZE_CREATE_OPERATOR(ConvBias)

  52. MEGDNN_SPECIALIZE_CREATE_OPERATOR(ConvolutionBackwardData)

  53. 

  54. #pragma GCC diagnostic push

  55. #pragma GCC diagnostic ignored "-Wpragmas"

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

  57. MEGDNN_FOREACH_OPR_CLASS(MEGDNN_INST_CREATE_OPERATOR)

  58. #pragma GCC diagnostic pop

  59. 

  60. }  // namespace arm_common

  61. }  // namespace megdnn

  62. 

  63. // vim: syntax=cpp.doxygen