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.

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

[TVM4J] Upgrade support for NDArray Container #1482

Closed
tqchen opened this issue Jul 24, 2018 · 2 comments
Closed

[TVM4J] Upgrade support for NDArray Container #1482

tqchen opened this issue Jul 24, 2018 · 2 comments

Comments

@tqchen
Copy link
Member

tqchen commented Jul 24, 2018

See #1418

The change is pretty simple

  • TVMArrayAlloc now contains a field with type kNDArrayContainer
  • Support return value kNDArrayContainer and create an NDArray.

This enables us to get returned NDArray from PackedFunc

@tqchen
Copy link
Member Author

tqchen commented Jul 24, 2018

cc @yzhliu

@yzhliu
Copy link
Member

yzhliu commented Jul 25, 2018

I'll look into it tomorrow

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants