Skip to content

Commit 29ecc63

Browse files
committed
Imperative mode prototype
Signed-off-by: Michal Zientkiewicz <michalz@nvidia.com>
1 parent 74f92e0 commit 29ecc63

7 files changed

Lines changed: 2422 additions & 1 deletion

File tree

dali/python/nvidia/dali/experimental/dali2/__init__.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,13 @@
1717
"""
1818

1919
from ._eval_mode import EvalMode # noqa: F401
20-
from ._eval_context import EvalContext # noqa: F401
2120
from ._type import * # noqa: F401, F403
21+
from ._tensor import Tensor, tensor, as_tensor # noqa: F401
22+
from ._batch import Batch, batch, as_batch # noqa: F401
2223
from ._device import Device # noqa: F401
24+
25+
from . import _fn
26+
from . import ops
27+
28+
ops._initialize()
29+
_fn._initialize()

0 commit comments

Comments
 (0)