Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

TTTensor class #23089

Merged
merged 15 commits into from
Sep 17, 2023
Merged

TTTensor class #23089

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
1fc5ff2
Added tt_tensor.py
mobley-trent Sep 5, 2023
6653b7a
Updated ivy/__init__.py, ivy/data_classes/__init__.py
mobley-trent Sep 6, 2023
f1f770f
Added test_tt_tensor.py, Updated test_creation.py, creation.py, tt_te…
mobley-trent Sep 7, 2023
3198497
Updated testing strategy - test_creation.py
mobley-trent Sep 11, 2023
939899d
Updated testing strategy - test_creation.py
mobley-trent Sep 11, 2023
919a3e9
Merge branch 'tensor_train' of https://github.com/mobley-trent/ivy in…
mobley-trent Sep 11, 2023
180507c
Updated test_creation.py
mobley-trent Sep 11, 2023
a511ffe
Merge branch 'tensor_train' of https://github.com/mobley-trent/ivy in…
mobley-trent Sep 11, 2023
005e879
Updated test_creation.py
mobley-trent Sep 11, 2023
3cc6f6d
Updated creation.py, test_tt_tensor.py
mobley-trent Sep 11, 2023
7020c16
Updated test_tt_tensor.py
mobley-trent Sep 11, 2023
b1f3372
Updated demos
mobley-trent Sep 12, 2023
8f6711a
Merge branch 'main' of https://github.com/unifyai/ivy into tensor_train
mobley-trent Sep 12, 2023
0adbe90
Updated test_creation.py
mobley-trent Sep 14, 2023
605ede3
Updated test_creation.py, test_tt_tensor.py
mobley-trent Sep 17, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion ivy/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -83,6 +83,10 @@ class CPTensor:
pass


class TTTensor:
pass


class Device(str):
def __new__(cls, dev_str):
if dev_str != "":
Expand Down Expand Up @@ -758,7 +762,7 @@ class Node(str):
add_ivy_container_instance_methods,
)
from .data_classes.nested_array import NestedArray
from .data_classes.factorized_tensor import TuckerTensor, CPTensor
from .data_classes.factorized_tensor import TuckerTensor, CPTensor, TTTensor
from ivy.utils.backend import (
current_backend,
compiled_backends,
Expand Down
1 change: 1 addition & 0 deletions ivy/data_classes/factorized_tensor/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,2 +1,3 @@
from .tucker_tensor import TuckerTensor
from .cp_tensor import CPTensor
from .tt_tensor import TTTensor
Loading
Loading