1#ifndef HALIDE_CODEGEN_PYTORCH_H
2#define HALIDE_CODEGEN_PYTORCH_H
This header file defines operators that let you dump a Halide expression, statement,...
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.
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
Definition of a lowered function.