Searched refs:kPrimGeLU (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | gelu.cc | 59 REGISTER_PRIMITIVE_EVAL_IMPL(GeLU, prim::kPrimGeLU, GeLUInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | nonlinear_fuc_ops_declare.cc | 138 REG_ADPT_DESC(Gelu, prim::kPrimGeLU->name(), ADPT_DESC(Gelu))
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 57 {kAllTarget, OpLevel_0, prim::kPrimGeLU}, in GetExpandOps()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | arithmetic_self_cpu_kernel.cc | 318 {prim::kPrimGeLU->name(), Gelu<T>}, in LaunchKernel()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 363 inline const PrimitivePtr kPrimGeLU = std::make_shared<Primitive>(kGeLU); variable
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 714 if (cnode->IsApply(prim::kPrimGeLU)) { in ExportNodes()
|