Xuanlei Zhao 10c61de2f7 [autochunk] support vit (#3084) ... support vit for autochunk * support some new ops for vit * fix some bugs * add test for vit		2023-03-10 10:23:26 +08:00
..
autochunk_codegen.py	[autochunk] support vit (#3084)	2023-03-10 10:23:26 +08:00
estimate_memory.py	[autochunk] refactor chunk memory estimation (#2762)	2023-03-08 16:22:30 +08:00
reorder_graph.py	[autochunk] support multi outputs chunk search (#2538)	2023-02-01 13:18:51 +08:00
search_chunk.py	[autochunk] refactor chunk memory estimation (#2762)	2023-03-08 16:22:30 +08:00
select_chunk.py	[autochunk] refactor chunk memory estimation (#2762)	2023-03-08 16:22:30 +08:00
trace_flow.py	[autochunk] support diffusion for autochunk (#2621)	2023-02-07 16:32:45 +08:00
trace_indice.py	[autochunk] support vit (#3084)	2023-03-10 10:23:26 +08:00
utils.py	[autochunk] support vit (#3084)	2023-03-10 10:23:26 +08:00