 Jinjing Zhou
|
e9440acb06
|
[TF] TF backend fix and new logic to choose backend (#1393)
* TF backend fix and new logic to choose backend
* fix
* fix
* fix
* fix
* fix backend
* fix
* dlpack alignment
* add flag
* flag
* lint
* lint
* remove unused
* several fixes
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
|
2020-03-30 18:45:17 +08:00 |
|
|
Quan (Andy) Gan
|
1c4bfb62bb
|
[Performance] Replace np.array with np.asarray (#1301)
* replace np.array with np.asarray
* fix
Co-authored-by: Minjie Wang <minjie.wang@nyu.edu>
|
2020-03-03 20:45:41 +08:00 |
|
|
Quan (Andy) Gan
|
86fe58ebb8
|
[Bug] Another fix on PyTorch memory leakage issue (#1139)
* another fix
* another try
* fix
* rewriting with kernel functions
* revert mxnet softmax changes
* lint fix
Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com>
|
2020-01-14 22:54:19 +08:00 |
|
|
VoVAllen
|
d30a69bf27
|
[Backend] TF backend (#978)
* tf
* add builtin support
* fiix
* pytest
* fix
* fix
* fix some bugs
* fix selecting
* fix todo
* fix test
* fix test fail in tf
* fix
* fix
* fix gather row
* fix gather row
* log backend
* fix gather row
* fix gather row
* fix for pytorch
* fix
* fix
* fix
* fix
* fix
* fix tests
* fix
* fix
* fix
* fix
* fix
* fix
* fix convert
* fix
* fix
* fix
* fix inplace
* add alignment setting
* add debug option
* Revert "add alignment setting"
This reverts commit ec63fb3506ea84fff7d447a1fbdfd1d5d1fb6110.
* tf ci
* fix lint
* fix lint
* add tfdlpack
* fix type
* add env
* fix backend
* fix
* fix tests
* remove one_hot
* remove comment
* remove comment
* fix
* use pip to install all
* fix test
* fix base
* fix
* fix
* add skip
* upgrade cmake
* change version
* change ci
* fix
* fix
* fix
* fix
* fix seg fault
* fix
* fix python version
* fix
* try fix
* fix
* fix
* tf takes longer time in ci
* change py version
* fix
* fix
* fix oom
* change kg env
* change kg env
* 啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊
* 我再也不搞各种乱七八糟环境了……
* use pytest
* Chang image
|
2019-12-20 15:56:51 +08:00 |
|