doc_translator.h File Reference

#include <tvm/ffi/container/dict.h>
#include <tvm/ffi/container/list.h>
#include <tvm/ffi/container/map.h>
#include <tvm/ffi/expected.h>
#include <tvm/ffi/reflection/accessor.h>
#include <tvm/ir/expr.h>
#include <tvm/script/printer/doc.h>
#include <optional>
#include <utility>

Include dependency graph for doc_translator.h:

Go to the source code of this file.

Classes
struct	tvm::script::printer::DocTranslatorVTable
	Engine callbacks borrowed for the translator's lifetime. More...
class	tvm::script::printer::DocTranslatorObj
	Eager translation context shared by registered hooks. Hooks construct Docs, use explicit lexical scopes, and attach origins when emitting. More...
class	tvm::script::printer::DocTranslator
	Owning reference to a hook translation context. More...

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::script
namespace	tvm::script::printer
namespace	tvm::script::printer::type_attr
namespace	tvm::script::printer::op_attr

Typedefs
using	tvm::script::printer::FDocTranslate = ffi::reflection::NativeFunctionView< ffi::Optional< ExprDoc >(DocTranslatorObj *translator, ffi::AnyView value, const ffi::Object *destination)>
	Native translation hook with borrowed inputs and an owning result.

Functions
ffi::Array< StmtDoc >	tvm::script::printer::ToStmtDocArray (const ffi::List< Doc > &docs)
	Convert collected Docs to statements in their original order.
TVM_TIRX_DLL Doc	tvm::script::printer::DocTranslate (ffi::AnyView ir, ffi::Dict< Doc, ffi::ObjectRef > *doc_origins=nullptr, ffi::Map< ffi::String, ffi::Any > extra_config={})
	Translate IR into a complete Doc including declarations and metadata setup.

Variables
constexpr const char *	tvm::script::printer::type_attr::kDocTranslate = "__tvm_doc_translate__"
	Type attribute containing a native or packed translation hook.
constexpr const char *	tvm::script::printer::op_attr::kDocTranslateOpCall = "__tvm_doc_translate_op_call__"
	Op-specific Call translation hook with the FDocTranslate contract.