1 #ifndef HALIDE_CODEGEN_PYTORCH_H 2 #define HALIDE_CODEGEN_PYTORCH_H 44 #endif // HALIDE_CODEGEN_PYTORCH_H This class emits C++ code to wrap a Halide pipeline so that it can be used as a C++ extension operato...
This header file defines operators that let you dump a Halide expression, statement, or type directly into an output stream in a human readable form.
This file defines the class FunctionDAG, which is our representation of a Halide pipeline, and contains methods to using Halide's bounds tools to query properties of it.
An IRVisitor that emits IR to the given output stream in a human readable form.
CodeGen_PyTorch(std::ostream &dest)
void compile(const Module &module)
Emit the PyTorch C++ wrapper for the Halide pipeline.
Definition of a lowered function.
Not visible externally, similar to 'static' linkage in C.
~CodeGen_PyTorch() override=default