def fused_bn_update_grad(poly_sch, fuzz_shape=None): test_fused_bn_update_grad((256, 56, 56, 256), (256, ), layout="NHWC", poly_sch=poly_sch)
def test_ms_fused_bn_update_grad(): test_fused_bn_update_grad((256, 56, 56, 256), (256, ), layout="NHWC", poly_sch=True) return True