Files
dmlc--dgl/python/dgl/backend/tensorflow/__init__.py
T
2026-07-13 13:35:51 +08:00

7 lines
105 B
Python

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