예제 #1
0
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)
예제 #2
0
def test_ms_fused_bn_update_grad():
    test_fused_bn_update_grad((256, 56, 56, 256), (256, ),
                              layout="NHWC",
                              poly_sch=True)
    return True