| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_aten.cpp | 36 SimpleIREvaluator ir_eval(stmt, {a_buf, b_buf}); in TEST() local
63 SimpleIREvaluator ir_eval(stmt, {a_buf, b_buf}); in TEST() local
90 SimpleIREvaluator ir_eval(stmt, {a_buf, b_buf}); in TEST() local
124 SimpleIREvaluator ir_eval(stmt, {a_buf, b_buf, c_buf, d_buf}); in TEST() local
160 SimpleIREvaluator ir_eval(stmt, {a_buf, b_buf, c_buf, d_buf}); in TEST() local
196 SimpleIREvaluator ir_eval(stmt, {a_buf, b_buf, c_buf, d_buf}); in TEST() local
232 SimpleIREvaluator ir_eval(stmt, {a_buf, b_buf, c_buf, d_buf}); in TEST() local
268 SimpleIREvaluator ir_eval(stmt, {a_buf, b_buf, c_buf, d_buf}); in TEST() local
308 SimpleIREvaluator ir_eval(stmt, {a_buf, b_buf, c_buf, d_buf, e_buf}); in TEST() local
349 SimpleIREvaluator ir_eval(stmt, {a_buf, b_buf, c_buf, d_buf, e_buf}); in TEST() local
[all …]
|
| H A D | test_external_calls.cpp | 80 SimpleIREvaluator ir_eval(l.root_stmt(), {Input, Weight, Bias, Result}); in TEST() local
134 SimpleIREvaluator ir_eval(l.root_stmt(), {Input, Weight, Bias, Result}); in TEST() local
175 SimpleIREvaluator ir_eval(l.root_stmt(), {Input, Weight, Result}); in TEST() local
234 SimpleIREvaluator ir_eval(l.root_stmt(), {Input, Weight, Bias, Result}); in TEST() local
295 SimpleIREvaluator ir_eval(l.root_stmt(), {Input, Weight, Bias, Result}); in TEST() local
336 SimpleIREvaluator ir_eval(l.root_stmt(), {Input, Weight, Result}); in TEST() local
383 SimpleIREvaluator ir_eval(l.root_stmt(), {Input, Mat1, Mat2, Result}); in TEST() local
433 SimpleIREvaluator ir_eval(l.root_stmt(), {Weight, Indices, Result}); in TEST() local
476 SimpleIREvaluator ir_eval(l.root_stmt(), {Input, Result}); in TEST() local
541 SimpleIREvaluator ir_eval(l.root_stmt(), {Input, DummyPrepacked, Result}); in TEST() local
[all …]
|
| H A D | test_expr.cpp | 324 SimpleIREvaluator ir_eval(stmt, {a_buf, b_buf, c_buf}); in TEST() local
349 SimpleIREvaluator ir_eval(memcpy_expr, {a, b, c}); in TEST() local
393 SimpleIREvaluator ir_eval(select_expr, {a, b, c}); in TEST() local
416 SimpleIREvaluator ir_eval(abs_expr, {a, b}); in TEST() local
|
| H A D | tutorial.cpp | 396 SimpleIREvaluator ir_eval(loopnest.root_stmt(), {A, B, X}); in main() local
|
| H A D | test_loopnest.cpp | 120 SimpleIREvaluator ir_eval(stmt, {tensor}); in TEST() local
564 SimpleIREvaluator ir_eval(stmt, {tensor}); in TEST() local
932 SimpleIREvaluator ir_eval(stmt, {a_buf, b_buf, c}); in TEST() local
|
| H A D | test_llvm.cpp | 249 LLVMCodeGen ir_eval(stmt, {a_buf, b_buf}); in TEST() local
|