transform.h File Reference

The transform passes for TIR-based Unified Static Memory Planner. More...

#include <tvm/tir/usmp/utils.h>

Include dependency graph for transform.h:

Go to the source code of this file.

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::tir
	tvm::tir::usmp
	tvm::tir::usmp::transform

Typedefs
using	tvm::tir::usmp::transform::Pass = tvm::transform::Pass

Functions
Pass	tvm::tir::usmp::transform::ConvertPoolAllocationsToOffsets (const Map< tir::Stmt, PoolAllocation > &pool_allocations, Bool emit_tvmscript_printable=Bool(false))
	Convert the analyzed PoolAllocation to offsets from pool variables. More...
Pass	tvm::tir::usmp::transform::AssignPoolInfo ()
	Assign PoolInfo objects to tir.allocate nodes depending on the PrimFunc's target. More...
Pass	tvm::tir::usmp::transform::CreateAllocatesForIO ()
	This pass creates Allocate nodes for I/O tensors. More...

Detailed Description

The transform passes for TIR-based Unified Static Memory Planner.