Searched refs:CppPrinter (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/test/cpp/tensorexpr/ |
| D | test_cpp_codegen.cpp | 19 CppPrinter printer(&ss); \
25 CppPrinter printer(&ss); \
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
[all …]
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | cpp_codegen.cpp | 46 CppPrinter::CppPrinter(std::ostream* os) : IRPrinter(*os), lane_(0) {} in CppPrinter() function in torch::jit::tensorexpr::CppPrinter
48 CppPrinter::~CppPrinter() = default;
50 void CppPrinter::printPrologue() { in printPrologue()
158 void CppPrinter::visit(const RampPtr& v) { in visit()
162 void CppPrinter::visit(const BroadcastPtr& v) { in visit()
166 void CppPrinter::visit(const ModPtr& v) { in visit()
170 void CppPrinter::visit(const MaxPtr& v) { in visit()
174 void CppPrinter::visit(const MinPtr& v) { in visit()
178 void CppPrinter::visit(const CompareSelectPtr& v) { in visit()
184 void CppPrinter::visit(const IfThenElsePtr& v) { in visit()
[all …]
|
| D | cpp_codegen.h | 19 class TORCH_API CppPrinter : public IRPrinter {
21 explicit CppPrinter(std::ostream* os);
22 ~CppPrinter() override;
94 std::unique_ptr<CppPrinter> printer_;
|
| /external/pytorch/torch/_inductor/codegen/rocm/ |
| D | rocm_kernel.py | 8 from ..cpp_utils import CppPrinter
18 cexpr = CppPrinter().doprint
|
| /external/pytorch/torch/_inductor/codegen/cuda/ |
| D | cuda_kernel.py | 18 from ..cpp_utils import CppPrinter, DTYPE_TO_CPP
26 cexpr = CppPrinter().doprint
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | cpp_utils.py | 225 class CppPrinter(ExprPrinter): class
434 cexpr = CppPrinter().doprint
|