mirrors / dmlc--dgl

项目文件夹

文件

T

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

5 行

82 B

Python

原始文件 永久链接 Blame 文件历史

 import os
 os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
 from .tensor import *