def fused_bn_reduce_grad(poly_sch, fuzz_shape=None): test_fused_bn_reduce_grad((256, 56, 56, 256), layout='NHWC', poly_sch=poly_sch)
def test_ms_fused_bn_reduce_grad(): test_fused_bn_reduce_grad((256, 56, 56, 256), layout='NHWC', poly_sch=True) return True