提交 189f88fb 编写于 作者: C chenzomi

change the hswish to HSwish

上级 3d3b9d54
...@@ -12,16 +12,16 @@ ...@@ -12,16 +12,16 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
"""hswish""" """HSwish"""
import _akg.topi as topi import _akg.topi as topi
import _akg.tvm as tvm import _akg.tvm as tvm
from _akg.topi import tag from _akg.topi import tag
@tvm.tag_scope(tag=tag.ELEMWISE) @tvm.tag_scope(tag=tag.ELEMWISE)
def topi_nn_hswish(x): def topi_nn_HSwish(x):
""" """
topi hswish topi HSwish
Args: Args:
x: x:
...@@ -42,7 +42,7 @@ def HSwish(x): ...@@ -42,7 +42,7 @@ def HSwish(x):
Returns: Returns:
""" """
return topi_nn_hswish(x) return topi_nn_HSwish(x)
def gpu_schedule_HSwish(outs): def gpu_schedule_HSwish(outs):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册