Browse Source

!14626 Fix Gelu in select ops

From: @liangzhibo
Reviewed-by: @ginfung,@zh_qh
Signed-off-by: @zh_qh
tags/v1.2.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
934669f947
2 changed files with 4 additions and 4 deletions
  1. +2
    -2
      mindspore/nn/layer/activation.py
  2. +2
    -2
      mindspore/ops/_selected_ops.py

+ 2
- 2
mindspore/nn/layer/activation.py View File

@@ -415,7 +415,7 @@ class GELU(Cell):

def __init__(self):
super(GELU, self).__init__()
self.gelu = _selected_ops.Gelu()
self.gelu = _selected_ops.GeLU()

def construct(self, x):
return self.gelu(x)
@@ -458,7 +458,7 @@ class FastGelu(Cell):

def __init__(self):
super(FastGelu, self).__init__()
self.fast_gelu = _selected_ops.FastGelu()
self.fast_gelu = _selected_ops.FastGeLU()

def construct(self, x):
return self.fast_gelu(x)


+ 2
- 2
mindspore/ops/_selected_ops.py View File

@@ -73,13 +73,13 @@ class Tanh:


@op_selector
class Gelu:
class GeLU:
def __call__(self, *args):
pass


@op_selector
class FastGelu:
class FastGeLU:
def __call__(self, *args):
pass



Loading…
Cancel
Save