yystopf
/
mindspore
forked from mindspore-Ecosystem/mindspore
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
45,876 Commits 23 Branches 0 Tags 3.9 GiB
Commit Graph

11 Commits

Author SHA1 Message Date
liqiliang 4be416f40f Add a gpu kernel, MatrixBandPart. 2022-05-14 10:46:38 +08:00
z00512249 fb1b4324d7 refactor matrix_set_diag_v3 for gpu backend. 2022-05-11 14:34:21 +08:00
liqiliang a67193ccb4 Add a cpu kernel, MatrixBandPart. 2022-05-10 14:51:53 +08:00
z00512249 528e957d92 opt test matrix_set_diag test codes 2022-01-29 15:13:08 +08:00
wenbean 860f52e45d Add ops warpper and test, add parralel launch 2022-01-29 11:38:30 +08:00
wenbean 28ec747fe5 Add GPU matrix_band_part 2022-01-27 17:22:09 +08:00
wenbean 3ab08cac55 Add ops warpper and test, add parralel launch 2022-01-26 14:42:50 +08:00
z00512249 cffcb7c74a add matrix_set_diag kernel for backend gpu 2022-01-25 21:15:21 +08:00
z00512249 4fddf02b08 add cholesky grad for backend cpu. 2022-01-24 17:32:02 +08:00
wenbean d700b5daa7 Add matrix_band_part cpu kernel and test cases 2022-01-20 09:46:33 +08:00
z00512249 815e49910e add kernel matrix_set_diag of backend cpu 2022-01-17 15:29:18 +08:00