tony/tensorneat-mend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
40 Commits 2 Branches 0 Tags
9b56f4ff73a2972ece5b11eaf13c1330e72978b1
Commit Graph

1 Commits

Author SHA1 Message Date
wls2002
9b56f4ff73 做了一些时间测试： 
1. jit中的vmap一个函数不会触发重新编译
2. jax.lax.while_lop单独执行确实可以提前中断，但在vmap中的性能需要考虑。
 2023-05-12 16:42:57 +08:00