-
Notifications
You must be signed in to change notification settings - Fork 2
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.
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
Adds Tensors objects to DLA #54
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Some ideas for improvements to avoid future stumbles.
Tested suggested changes with #51 and everything seems to work |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ideally, replace unwrap
with unwrap_unchecked
where fitting but please merge afterwards.
#51 Turned out to be way too big to review, so I split tensor functionality to this PR to reduce the amount of code to review.
Tensors are used to abstract the shape of data. This is useful since DLA expected a different data shape than what TVM does. Also since different models might use other data formats it is sensible to generalize tensors to all possible data shapes.