4 Commits

Author	SHA1	Message	Date
liqiliang	e0369d0d54	Extend the MatrixBandPart kernel to support batch.	2022-06-08 20:45:19 +08:00
liqiliang	d200f89202	Add dynamic_shape tests.	2022-06-01 17:58:33 +08:00
liqiliang	3f1c71e04a	Add vmap and it's tests.	2022-05-31 09:26:13 +08:00
liqiliang	4be416f40f	Add a gpu kernel, MatrixBandPart.	2022-05-14 10:46:38 +08:00