def fused_bn_reduce(poly_sch, fuzz_shape=None): test_fused_bn_reduce((256, 7, 7, 2048), layout='NHWC', poly_sch=poly_sch)
def test_ms_fused_bn_reduce(): test_fused_bn_reduce((256, 7, 7, 2048), layout='NHWC', poly_sch=True) return True