optimize_assert (reference) in projects: aosp_15_r20

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched refs:optimize_assert (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/test/dynamo/
H A D	test_skip_non_tensor.py	`73 opt_fn = torch._dynamo.optimize_assert(counter)(fn) 86 opt_fn = torch._dynamo.optimize_assert(counter)(fn) 98 opt_fn = torch._dynamo.optimize_assert(counter)(fn) 110 opt_fn = torch._dynamo.optimize_assert(counter)(fn) 120 opt_fn = torch._dynamo.optimize_assert(counter)(fn) 136 with torch._dynamo.optimize_assert(counter): 158 with torch._dynamo.optimize_assert(counter):`
H A D	test_python_autograd.py	`207 fn_dynamo = torch._dynamo.optimize_assert(cnt)(fn) 274 opt_forward = torch._dynamo.optimize_assert(cnt)(forward) 275 opt_grad = torch._dynamo.optimize_assert(cnt)(grad)`
H A D	test_base_output.py	`42 opt_fn = torch._dynamo.optimize_assert(cnts)(fn) 56 opt_fn = torch._dynamo.optimize_assert(cnts)(fn)`
H A D	test_model_output.py	`87 opt_fn = torch._dynamo.optimize_assert(cnts)(fn) 103 opt_fn = torch._dynamo.optimize_assert(cnts)(fn)`
H A D	test_backends.py	`63 opt_fn = torch._dynamo.optimize_assert(compiler_fn)(fn) 95 opt_fn = torch._dynamo.optimize_assert(compiler_fn)(fn)`
H A D	test_verify_correctness.py	`79 opt_fn = torch._dynamo.optimize_assert(compiler_fn)(fn)`
H A D	test_repros.py	1003 opt_fn = torch._dynamo.optimize_assert(cnt)(fn) 1230 opt_fn = torch._dynamo.optimize_assert(cnt)(fn) 1252 opt_model = torch._dynamo.optimize_assert(cnt)(model) 1336 opt_model = torch._dynamo.optimize_assert(cnt)(model) 1405 opt_fn = torch._dynamo.optimize_assert(cnt)(fn) 1426 opt_fn = torch._dynamo.optimize_assert(cnt)(fn) 1488 opt_model = torch._dynamo.optimize_assert(cnt)(model) 1507 opt_model = torch._dynamo.optimize_assert(cnt)(model) 1522 opt_fn = torch._dynamo.optimize_assert(cnt)(fn) 1607 opt_fn = torch._dynamo.optimize_assert(cnt)(fn) [all …]
H A D	test_misc.py	205 opt_boolarg = torch._dynamo.optimize_assert(counter)(boolarg) 302 opt_call_packed = torch._dynamo.optimize_assert(counter)(call_packed) 3353 opt_fn = torch._dynamo.optimize_assert(cnts)(fn) 3354 opt_fn_inner = torch._dynamo.optimize_assert(cnts)(opt_fn(x1)) 3382 opt_fn1 = torch._dynamo.optimize_assert(cnts)(fn1) 3383 tmp1 = torch._dynamo.optimize_assert(cnts)(opt_fn1()) 3384 tmp2 = torch._dynamo.optimize_assert(cnts)(opt_fn1()) 3417 opt_fn1 = torch._dynamo.optimize_assert(cnts)(fn1) 3418 counter2 = torch._dynamo.optimize_assert(cnts)(opt_fn1()) 3588 opt_fn = torch._dynamo.optimize_assert(cnts)(fn) [all …]
H A D	test_modules.py	`1253 opt_m3 = torch._dynamo.optimize_assert(cnt)(m3) 1254 opt_m4 = torch._dynamo.optimize_assert(cnt)(m4) 1270 with torch._dynamo.optimize_assert(cnt): 1277 with torch._dynamo.optimize_assert(cnt):`
H A D	test_functions.py	`1278 opt_fn = torch._dynamo.optimize_assert("eager")(fn) 1341 opt_fn = torch._dynamo.optimize_assert("eager")(fn) 1946 cfunc = torch._dynamo.optimize_assert(cnt, dynamic=dynamic)(func) 2023 cfunc = torch._dynamo.optimize_assert(cnt)(func)`
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A D	testing.py	`19 from . import config, eval_frame, optimize_assert, reset 258 opt_fn = optimize_assert(actual)(fn)`
H A D	__init__.py	`27 optimize_assert,`
H A D	eval_frame.py	`776 return optimize_assert( 1421 opt_f = optimize_assert( 1602 def optimize_assert( function`
/aosp_15_r20/external/pytorch/test/inductor/
H A D	test_torchinductor.py	6109 opt_fn = torch._dynamo.optimize_assert(compile_fx)(fn) 6130 opt_fn = torch._dynamo.optimize_assert(compile_fx)(fn) 6150 opt_fn = torch._dynamo.optimize_assert(compile_fx)(fn) 6164 opt_fn = torch._dynamo.optimize_assert(compile_fx)(fn) 6208 opt_fn = torch._dynamo.optimize_assert(compile_fx)(fn) 6216 opt_fn = torch._dynamo.optimize_assert(compile_fx)(fn) 7972 @torch._dynamo.optimize_assert("inductor") 7987 @torch._dynamo.optimize_assert("inductor") 8044 @torch._dynamo.optimize_assert("inductor") 8064 @torch._dynamo.optimize_assert("inductor") [all …]