tvm_tirx.backend.trn

Trainium-owned TIRx modules.

Functions

register_backend()	Register Trainium-owned Python semantics.
script_namespace()	Return the Trainium TVMScript namespace object.
script_namespaces(**_)	Return Trainium-owned TVMScript namespaces.

Namespaces

• tvm_tirx.backend.trn.op

• tvm_tirx.backend.trn.transform

• tvm_tirx.backend.trn.axis

• tvm_tirx.backend.trn.layout

• tvm_tirx.backend.trn.pipeline

• tvm_tirx.backend.trn.script

• tvm_tirx.backend.trn.thread_scope