diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-24 13:57:42 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-24 13:57:42 +0200 |
commit | d258c7d6936410cd78189445a0260d983f7684d6 (patch) | |
tree | 415b2fc393c68204055ff334882b08349e265507 /ot/gpu/__init__.py | |
parent | c9b99df8fffec1dcc6802ef43b6192774817c5fb (diff) |
convert ot.gpu to cupy
Diffstat (limited to 'ot/gpu/__init__.py')
-rw-r--r-- | ot/gpu/__init__.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/ot/gpu/__init__.py b/ot/gpu/__init__.py index a2fdd3d..de4825d 100644 --- a/ot/gpu/__init__.py +++ b/ot/gpu/__init__.py @@ -4,9 +4,13 @@ from . import bregman from . import da from .bregman import sinkhorn +from . import utils +from .utils import dist, to_gpu, to_np + + # Author: Remi Flamary <remi.flamary@unice.fr> # Leo Gautheron <https://github.com/aje> # # License: MIT License -__all__ = ["bregman", "da", "sinkhorn"] +__all__ = ["utils", "dist", "sinkhorn"] |