API reference

PyTorch-style reference for torch_tvarant. Each entry is a dedicated page with signature, parameters, and examples.

torch.tvarant

Device module — torch.tvarant overview

• is_available · device_count · backend
• set_device · synchronize
• manual_seed · get_rng_state · …

torch_tvarant.compiler

• compile · compile_fx
• trace_module · register
• Compiler overview

Supported ops

• Ops index — relu, add, mm, linear, linear_act, …
• Kernels — host / OpenCL catalog
• C++ — extension layout & TORCH_LIBRARY_IMPL