├── .gitignore ├── LICENSE ├── README.md ├── dataclasses_tensor ├── __init__.py ├── adapters.py ├── core.py ├── layout.py └── utils.py ├── setup.py └── tests ├── __init__.py ├── test_array.py ├── test_batch.py ├── test_dataclass.py ├── test_dtype.py ├── test_enum.py ├── test_optional.py ├── test_primitives.py └── test_union.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/README.md -------------------------------------------------------------------------------- /dataclasses_tensor/__init__.py: -------------------------------------------------------------------------------- 1 | from .core import dataclass_tensor, config 2 | -------------------------------------------------------------------------------- /dataclasses_tensor/adapters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/dataclasses_tensor/adapters.py -------------------------------------------------------------------------------- /dataclasses_tensor/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/dataclasses_tensor/core.py -------------------------------------------------------------------------------- /dataclasses_tensor/layout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/dataclasses_tensor/layout.py -------------------------------------------------------------------------------- /dataclasses_tensor/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/dataclasses_tensor/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_array.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/tests/test_array.py -------------------------------------------------------------------------------- /tests/test_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/tests/test_batch.py -------------------------------------------------------------------------------- /tests/test_dataclass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/tests/test_dataclass.py -------------------------------------------------------------------------------- /tests/test_dtype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/tests/test_dtype.py -------------------------------------------------------------------------------- /tests/test_enum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/tests/test_enum.py -------------------------------------------------------------------------------- /tests/test_optional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/tests/test_optional.py -------------------------------------------------------------------------------- /tests/test_primitives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/tests/test_primitives.py -------------------------------------------------------------------------------- /tests/test_union.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kachayev/dataclasses-tensor/HEAD/tests/test_union.py --------------------------------------------------------------------------------