Masahiro Masuda fc0473885f [TIR] VNNI and ARM dot product intrinsic for tensorization (#10925)

2022-04-07 18:34:38 -07:00

..

analysis

[TIR][MetaSchedule] Estimate TIR FLOPs (#10782)

2022-03-25 09:57:16 -04:00

schedule

[TIR][Analysis] Add SuggestIndexMap for layout rewriting (#10732)

2022-03-25 11:19:46 -07:00

tensor_intrin

[TIR] VNNI and ARM dot product intrinsic for tensorization (#10925)

2022-04-07 18:34:38 -07:00

transform

[TE][TIR] Implement layout transformations, non-flat memory buffers (#9727)

2022-03-06 21:21:50 -05:00

usmp

[USMP] adding support for U2 and U3 usecases (#10193)

2022-02-14 09:54:06 +00:00

__init__.py

[TIR][Analysis] Add SuggestIndexMap for layout rewriting (#10732)

2022-03-25 11:19:46 -07:00

_ffi_api.py

[REFACTOR][PY] Establish tvm.tir

2020-02-13 20:24:01 -08:00

buffer.py

[TE][TIR] Implement layout transformations, non-flat memory buffers (#9727)

2022-03-06 21:21:50 -05:00

data_layout.py

[MyPy] Extend type checking and annotation for TIR (#8429)

2021-07-12 20:34:26 -04:00

expr.py

[TE] Promote substituted variable to iter_var's dtype (#10571)

2022-03-12 10:16:15 +09:00

function.py

[TIR][Analysis] Add SuggestIndexMap for layout rewriting (#10732)

2022-03-25 11:19:46 -07:00

generic.py

fix annotation of tir generic (#9119)

2021-09-27 21:29:45 +09:00

ir_builder.py

[TIR] Fix int32 vs int64 mismatch in For construct. (#10595)

2022-04-05 15:08:36 +09:00

op.py

[MyPy] Extend type checking and annotation for TIR (#8429)

2021-07-12 20:34:26 -04:00

stmt_functor.py

[MyPy] Extend type checking and annotation for TIR (#8429)

2021-07-12 20:34:26 -04:00

stmt.py

[TIR] Updated python docstring and parameter names for AllocateConst (#10602)

2022-03-15 06:20:02 +09:00