fix topi.nn.global_pool layout="NHWC" (#4656)
author戚海涛 <qihaitao007@163.com>
Fri, 10 Jan 2020 19:00:10 +0000 (03:00 +0800)
committerTianqi Chen <tqchen@users.noreply.github.com>
Fri, 10 Jan 2020 19:00:10 +0000 (11:00 -0800)
commit8d83da6b39b5e75afb3a69bd4011050de05bd78c
tree5c017afa1d24a377fe71f1828cd6f940406bcf25
parent07b45d958d4af91ec1bab66f6cf391d1ce12ddaf
fix topi.nn.global_pool layout="NHWC" (#4656)

* Update topi.cc

fix topi.nn.global_pool layout="NHWC"

* add topi.nn.global_pool layout=NHWC test
topi/src/topi.cc
topi/tests/python/test_topi_pooling.py