Searched refs:GELU (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_gelu_grad_sens.py | 20 from mindspore.nn import Cell, GELU
38 n = GELU()
82 n = GELU()
|
| D | test_gelu.py | 21 from mindspore.nn import GELU
28 n = GELU()
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | activation.py | 390 class GELU(Cell): class
432 super(GELU, self).__init__()
865 'gelu': GELU,
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/ |
| D | npu_converter_utils.h | 52 GELU = 15, enumerator
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/ |
| D | activation_mapper.cc | 34 {mindspore::GELU, std::make_shared<ops::GeLU>()}, in Mapper()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | gelu_fusion.cc | 37 gelu_prim->set_activation_type(mindspore::GELU); in CreateGeLUNode()
|
| /third_party/mindspore/tests/ut/python/nn/ |
| D | test_activation.py | 83 self.gelu = nn.GELU()
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | ops_types.fbs | 81 GELU = 19,
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_parse_graph.h | 72 {GELU, OperatorType::kRecReLU},
|
| /third_party/mindspore/mindspore/lite/tools/converter/import/ |
| D | primitive_adjust.cc | 151 … {ops::kNameGeLU, mindspore::GELU},
165 {kNameGeLUGrad, mindspore::GELU},
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.h | 74 GELU = 19, enumerator
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | ops_utils.h | 254 constexpr char GELU[] = "GeLU"; variable
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_auto_parallel.cc | 134 static const std::set<std::string> elementwise_op = {ACTIVATION, GELU, TANH, in IsElementWiseOperator()
162 …{MATMUL, TRANSPOSE, GELU, TANH, SOFTMAX, SUB, MUL, DIV, RESHAPE, GREATER, LOG_SOFTMAX, ACTIVATION,… in IsSplittableOperator()
|