Tristan Konolige fe299d7688 [TVMSCRIPT] Improve tvmscript type hints (#11654) ...

* [TVMSCRIPT] Improve tvmscript type hints

- Change numeric types to classes so they work as function arguments.
- Add var as a class.
- Add floordiv, index, and mod to PrimExpr.

* use Union

2022-06-09 17:45:36 -07:00

..

__init__.py

[TIR][USMP] adding the pass to convert to pool offsets (#9418)

2021-12-09 14:45:16 +00:00

__init__.pyi

[TVMSCRIPT] Improve tvmscript type hints (#11654)

2022-06-09 17:45:36 -07:00

intrin.py

[Metaschedule] Add demonstration of selectively tuning relay ops with TIR schedules (#10793)

2022-03-30 13:30:44 -07:00

node.py

[TVMScript] Represent ramp as index slice (#11308)

2022-05-16 08:33:16 +08:00

prim_func.py

[TVMScript] Script namespace changes (#9115)

2021-10-01 13:10:25 -07:00

scope_handler.py

[TVMScript] Represent ramp as index slice (#11308)

2022-05-16 08:33:16 +08:00

special_stmt.py

[TVMScript] Represent ramp as index slice (#11308)

2022-05-16 08:33:16 +08:00

ty.py

[TVMScript] Allow T.Buffer[] arg annotation to use int as shape (#11454)

2022-05-30 16:13:50 +09:00