change the hswish to HSwish

This commit is contained in:
chenzomi 2020-05-12 11:29:49 +08:00
parent 3d3b9d5474
commit 189f88fbe1
1 changed files with 4 additions and 4 deletions

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):