data_type_rewriter.h File Reference

Rewrite the data type of expressions. More...

#include <tvm/tir/stmt_functor.h>
#include <unordered_map>

Include dependency graph for data_type_rewriter.h:

Go to the source code of this file.

Classes
class	tvm::tir::DataTypeLegalizer
	Legalize the data types of expressions to make sure they are consistent with other parts of the program. More...
class	tvm::tir::IndexDataTypeRewriter
	Data type rewriter for buffer indices. More...
class	tvm::tir::IndexDataTypeNormalizer
	Normalize the data types of buffer shapes and indices to the same data type. More...

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

Detailed Description

Rewrite the data type of expressions.