forked from huawei/mindspore2022
change the hswish to HSwish
This commit is contained in:
parent
3d3b9d5474
commit
189f88fbe1
|
|
@ -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…
Reference in New Issue