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: a1e6720756
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'a1e6720756'
${ noResults }
MegEngine/dnn/src/common/elemwise/each_mode.inl

105 lines
4.4 kB
Raw Blame History 

  1. /**

  2.  * \file dnn/src/common/elemwise/each_mode.inl

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

  4.  *

  5.  * Copyright (c) 2014-2020 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. // generated by gen_elemwise_each_mode.py

  12. #define MEGDNN_FOREACH_ELEMWISE_MODE_UNARY_BOOL(cb) \

  13.     MEGDNN_ELEMWISE_MODE_ENABLE(NOT, cb) \

  14. 

  15. #define MEGDNN_FOREACH_ELEMWISE_MODE_UNARY_FLOAT(cb) \

  16.     MEGDNN_ELEMWISE_MODE_ENABLE(RELU, cb) \

  17.     MEGDNN_ELEMWISE_MODE_ENABLE(ABS, cb) \

  18.     MEGDNN_ELEMWISE_MODE_ENABLE(NEGATE, cb) \

  19.     MEGDNN_ELEMWISE_MODE_ENABLE(ACOS, cb) \

  20.     MEGDNN_ELEMWISE_MODE_ENABLE(ASIN, cb) \

  21.     MEGDNN_ELEMWISE_MODE_ENABLE(CEIL, cb) \

  22.     MEGDNN_ELEMWISE_MODE_ENABLE(COS, cb) \

  23.     MEGDNN_ELEMWISE_MODE_ENABLE(EXP, cb) \

  24.     MEGDNN_ELEMWISE_MODE_ENABLE(EXPM1, cb) \

  25.     MEGDNN_ELEMWISE_MODE_ENABLE(FLOOR, cb) \

  26.     MEGDNN_ELEMWISE_MODE_ENABLE(LOG, cb) \

  27.     MEGDNN_ELEMWISE_MODE_ENABLE(LOG1P, cb) \

  28.     MEGDNN_ELEMWISE_MODE_ENABLE(SIGMOID, cb) \

  29.     MEGDNN_ELEMWISE_MODE_ENABLE(SIN, cb) \

  30.     MEGDNN_ELEMWISE_MODE_ENABLE(TANH, cb) \

  31.     MEGDNN_ELEMWISE_MODE_ENABLE(FAST_TANH, cb) \

  32.     MEGDNN_ELEMWISE_MODE_ENABLE(ROUND, cb) \

  33.     MEGDNN_ELEMWISE_MODE_ENABLE(ERF, cb) \

  34.     MEGDNN_ELEMWISE_MODE_ENABLE(ERFINV, cb) \

  35.     MEGDNN_ELEMWISE_MODE_ENABLE(ERFC, cb) \

  36.     MEGDNN_ELEMWISE_MODE_ENABLE(ERFCINV, cb) \

  37.     MEGDNN_ELEMWISE_MODE_ENABLE(H_SWISH, cb) \

  38. 

  39. #define MEGDNN_FOREACH_ELEMWISE_MODE_UNARY_INT(cb) \

  40.     MEGDNN_ELEMWISE_MODE_ENABLE(RELU, cb) \

  41.     MEGDNN_ELEMWISE_MODE_ENABLE(ABS, cb) \

  42.     MEGDNN_ELEMWISE_MODE_ENABLE(NEGATE, cb) \

  43. 

  44. #define MEGDNN_FOREACH_ELEMWISE_MODE_BINARY_BOOL(cb) \

  45.     MEGDNN_ELEMWISE_MODE_ENABLE(AND, cb) \

  46.     MEGDNN_ELEMWISE_MODE_ENABLE(OR, cb) \

  47.     MEGDNN_ELEMWISE_MODE_ENABLE(XOR, cb) \

  48.     MEGDNN_ELEMWISE_MODE_ENABLE(LT, cb) \

  49.     MEGDNN_ELEMWISE_MODE_ENABLE(LEQ, cb) \

  50.     MEGDNN_ELEMWISE_MODE_ENABLE(EQ, cb) \

  51. 

  52. #define MEGDNN_FOREACH_ELEMWISE_MODE_BINARY_FLOAT(cb) \

  53.     MEGDNN_ELEMWISE_MODE_ENABLE(ABS_GRAD, cb) \

  54.     MEGDNN_ELEMWISE_MODE_ENABLE(ADD, cb) \

  55.     MEGDNN_ELEMWISE_MODE_ENABLE(FLOOR_DIV, cb) \

  56.     MEGDNN_ELEMWISE_MODE_ENABLE(MAX, cb) \

  57.     MEGDNN_ELEMWISE_MODE_ENABLE(MIN, cb) \

  58.     MEGDNN_ELEMWISE_MODE_ENABLE(MOD, cb) \

  59.     MEGDNN_ELEMWISE_MODE_ENABLE(MUL, cb) \

  60.     MEGDNN_ELEMWISE_MODE_ENABLE(SIGMOID_GRAD, cb) \

  61.     MEGDNN_ELEMWISE_MODE_ENABLE(SUB, cb) \

  62.     MEGDNN_ELEMWISE_MODE_ENABLE(SWITCH_GT0, cb) \

  63.     MEGDNN_ELEMWISE_MODE_ENABLE(TANH_GRAD, cb) \

  64.     MEGDNN_ELEMWISE_MODE_ENABLE(LT, cb) \

  65.     MEGDNN_ELEMWISE_MODE_ENABLE(LEQ, cb) \

  66.     MEGDNN_ELEMWISE_MODE_ENABLE(EQ, cb) \

  67.     MEGDNN_ELEMWISE_MODE_ENABLE(FUSE_ADD_RELU, cb) \

  68.     MEGDNN_ELEMWISE_MODE_ENABLE(TRUE_DIV, cb) \

  69.     MEGDNN_ELEMWISE_MODE_ENABLE(POW, cb) \

  70.     MEGDNN_ELEMWISE_MODE_ENABLE(LOG_SUM_EXP, cb) \

  71.     MEGDNN_ELEMWISE_MODE_ENABLE(FUSE_ADD_TANH, cb) \

  72.     MEGDNN_ELEMWISE_MODE_ENABLE(FAST_TANH_GRAD, cb) \

  73.     MEGDNN_ELEMWISE_MODE_ENABLE(FUSE_ADD_SIGMOID, cb) \

  74.     MEGDNN_ELEMWISE_MODE_ENABLE(ATAN2, cb) \

  75.     MEGDNN_ELEMWISE_MODE_ENABLE(H_SWISH_GRAD, cb) \

  76.     MEGDNN_ELEMWISE_MODE_ENABLE(FUSE_ADD_H_SWISH, cb) \

  77. 

  78. #define MEGDNN_FOREACH_ELEMWISE_MODE_BINARY_INT(cb) \

  79.     MEGDNN_ELEMWISE_MODE_ENABLE(ABS_GRAD, cb) \

  80.     MEGDNN_ELEMWISE_MODE_ENABLE(ADD, cb) \

  81.     MEGDNN_ELEMWISE_MODE_ENABLE(FLOOR_DIV, cb) \

  82.     MEGDNN_ELEMWISE_MODE_ENABLE(MAX, cb) \

  83.     MEGDNN_ELEMWISE_MODE_ENABLE(MIN, cb) \

  84.     MEGDNN_ELEMWISE_MODE_ENABLE(MOD, cb) \

  85.     MEGDNN_ELEMWISE_MODE_ENABLE(MUL, cb) \

  86.     MEGDNN_ELEMWISE_MODE_ENABLE(SIGMOID_GRAD, cb) \

  87.     MEGDNN_ELEMWISE_MODE_ENABLE(SUB, cb) \

  88.     MEGDNN_ELEMWISE_MODE_ENABLE(SWITCH_GT0, cb) \

  89.     MEGDNN_ELEMWISE_MODE_ENABLE(TANH_GRAD, cb) \

  90.     MEGDNN_ELEMWISE_MODE_ENABLE(LT, cb) \

  91.     MEGDNN_ELEMWISE_MODE_ENABLE(LEQ, cb) \

  92.     MEGDNN_ELEMWISE_MODE_ENABLE(EQ, cb) \

  93.     MEGDNN_ELEMWISE_MODE_ENABLE(FUSE_ADD_RELU, cb) \

  94.     MEGDNN_ELEMWISE_MODE_ENABLE(SHL, cb) \

  95.     MEGDNN_ELEMWISE_MODE_ENABLE(SHR, cb) \

  96.     MEGDNN_ELEMWISE_MODE_ENABLE(RMULH, cb) \

  97. 

  98. #define MEGDNN_FOREACH_ELEMWISE_MODE_TERNARY_FLOAT(cb) \

  99.     MEGDNN_ELEMWISE_MODE_ENABLE(COND_LEQ_MOV, cb) \

  100.     MEGDNN_ELEMWISE_MODE_ENABLE(FUSE_MUL_ADD3, cb) \

  101. 

  102. #define MEGDNN_FOREACH_ELEMWISE_MODE_TERNARY_INT(cb) \

  103.     MEGDNN_ELEMWISE_MODE_ENABLE(COND_LEQ_MOV, cb) \