|
|
@@ -1,13 +1,3 @@ |
|
|
|
/** |
|
|
|
* \file dnn/src/common/elemwise/each_mode.inl |
|
|
|
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License") |
|
|
|
* |
|
|
|
* Copyright (c) 2014-2020 Megvii Inc. All rights reserved. |
|
|
|
* |
|
|
|
* Unless required by applicable law or agreed to in writing, |
|
|
|
* software distributed under the License is distributed on an |
|
|
|
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
|
|
*/ |
|
|
|
// generated by gen_elemwise_each_mode.py |
|
|
|
#define MEGDNN_FOREACH_ELEMWISE_MODE_UNARY_BOOL(cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(NOT, cb) \ |
|
|
@@ -95,6 +85,8 @@ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(SHR, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(RMULH, cb) \ |
|
|
|
|
|
|
|
#define MEGDNN_FOREACH_ELEMWISE_MODE_TERNARY_BOOL(cb) \ |
|
|
|
|
|
|
|
#define MEGDNN_FOREACH_ELEMWISE_MODE_TERNARY_FLOAT(cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(COND_LEQ_MOV, cb) \ |
|
|
|
MEGDNN_ELEMWISE_MODE_ENABLE(FUSE_MUL_ADD3, cb) \ |
|
|
|