Halide 18.0.0
Halide compiler and libraries
|
This class emits C++ code to wrap a Halide pipeline so that it can be used as a C++ extension operator in PyTorch. More...
#include <CodeGen_PyTorch.h>
Public Member Functions | |
CodeGen_PyTorch (std::ostream &dest) | |
~CodeGen_PyTorch () override=default | |
void | compile (const Module &module) |
Emit the PyTorch C++ wrapper for the Halide pipeline. | |
![]() | |
IRPrinter (std::ostream &) | |
Construct an IRPrinter pointed at a given output stream (e.g. | |
void | print (const Expr &) |
emit an expression on the output stream | |
void | print_no_parens (const Expr &) |
Emit an expression on the output stream without enclosing parens. | |
void | print (const Stmt &) |
emit a statement on the output stream | |
void | print_list (const std::vector< Expr > &exprs) |
emit a comma delimited list of exprs, without any leading or trailing punctuation. | |
![]() | |
IRVisitor ()=default | |
virtual | ~IRVisitor ()=default |
This class emits C++ code to wrap a Halide pipeline so that it can be used as a C++ extension operator in PyTorch.
Definition at line 29 of file CodeGen_PyTorch.h.
Halide::Internal::CodeGen_PyTorch::CodeGen_PyTorch | ( | std::ostream & | dest | ) |
|
overridedefault |