Searched defs:CppPrinter (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_cpp_codegen.cpp | 29 TEST(CppPrinter, IntImm) { in TEST() argument
34 TEST(CppPrinter, FloatImm) { in TEST() argument
39 TEST(CppPrinter, FloatImm1) { in TEST() argument
44 TEST(CppPrinter, DoubleImm) { in TEST() argument
49 TEST(CppPrinter, DoubleImm1) { in TEST() argument
54 TEST(CppPrinter, HalfImm) { in TEST() argument
59 TEST(CppPrinter, Add) { in TEST() argument
64 TEST(CppPrinter, AddExpr1) { in TEST() argument
71 TEST(CppPrinter, AddExpr2) { in TEST() argument
78 TEST(CppPrinter, AddExpr3) { in TEST() argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | cpp_codegen.cpp | 46 CppPrinter::CppPrinter(std::ostream* os) : IRPrinter(*os), lane_(0) {} in CppPrinter() function in torch::jit::tensorexpr::CppPrinter
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_utils.py | 225 class CppPrinter(ExprPrinter): class
|