test_topi_pooling.py
18 KB
-
[Topi][Cuda]Optimizations of global_ave_pool for NHWC layout (#5450) · 0a1e1601
* Optimizations of global_ave_pool for NHWC layout * Optimize the code format to pass inspection of pylint Co-authored-by: Shawn-Inspur <wushaohua@inspur.com>
SXM-inspur committed