コード例 #1
0
def rsqrt(poly_sch, fuzz_shape=None):
    test_ms_rsqrt((32, 1024, 1024), 'float32', poly_sch=poly_sch)
    test_ms_rsqrt((32, 1024, 1024), 'float16', poly_sch=poly_sch)
コード例 #2
0
def test_rsqrt():
    test_ms_rsqrt((32, 1024, 1024), 'float32', poly_sch=True)
    test_ms_rsqrt((32, 1024, 1024), 'float16', poly_sch=True)
    return True