test_topi_depthwise_conv2d.py
10.4 KB
-
[TOPI] Depth wise convolution backward methods for NHWC (#434) · ffff1e49
* rename the nchw and pass the unit test; going to do it for nhwc depthwise * bug with fusion * nchw works fine; nhwc float32 problem remains * still cannot bind them together * fusion works * syntax fix * all bugs fixed; test cases pass * minor fix on nn.h * back wrt input * backward wrt input nhwc; only test case in recipe * test case for depthwise back wrt input * test case for depthwise backward wrt weight * tags * minor fixes * pylint test; add arch=3.7 * modify scheduler * better backward depthwise w.r.t weight scheduler * updated scheduler * test_topi_depthwise_conv2d_back_input.py and test_topi_depthwise_conv2d_back_weight.py success * all test cases wrt input pass * update * new test cases and scheduler * not working 1 and 2 * good wrt weight, bad wrt input * test cases added * remove tf lines * minor fix * compute arch changed * remove compile hook * minor change * pylint * fix the float for python case * fix cases for python3 case * except for memoize * fix most; memoize still wrong * memoize added * unexpected layout cases added for scheduler * error message layout other than NHWC added * improve padding * fix as pr requests * remove dilate in backward wrt weight
wetliu committed