Browse Source

change the hswish to HSwish

tags/v0.3.0-alpha
chenzomi 6 years ago
parent
commit
189f88fbe1
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      mindspore/_akg/gpu/hswish.py

+ 4
- 4
mindspore/_akg/gpu/hswish.py View File

@@ -12,16 +12,16 @@
# See the License for the specific language governing permissions and
# limitations under the License.

"""hswish"""
"""HSwish"""
import _akg.topi as topi
import _akg.tvm as tvm
from _akg.topi import tag


@tvm.tag_scope(tag=tag.ELEMWISE)
def topi_nn_hswish(x):
def topi_nn_HSwish(x):
"""
topi hswish
topi HSwish
Args:
x:

@@ -42,7 +42,7 @@ def HSwish(x):
Returns:

"""
return topi_nn_hswish(x)
return topi_nn_HSwish(x)


def gpu_schedule_HSwish(outs):


Loading…
Cancel
Save