Skip to content

Moving TF backend to eager mode and dlpack #565

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

Draft
wants to merge 12 commits into
base: master
Choose a base branch
from
Draft

Moving TF backend to eager mode and dlpack #565

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
545494a
Move to TFEager and dlpack
lantiga Mar 11, 2021
e92964b
Patch c_api_experimental for CI
lantiga Mar 11, 2021
f2e9559
Fix path
lantiga Mar 11, 2021
0a332ad
fix invalid access after freeing `next_item`
Mar 12, 2021
2a5229e
Fix double free issue when TFE manages memory
lantiga Mar 13, 2021
63f8ba0
Handle errors
lantiga Mar 13, 2021
4c48a92
Merge branch 'master' into tf-eager-dlpack
lantiga Mar 14, 2021
e1a7f02
Merge branch 'master' into tf-eager-dlpack
lantiga Mar 14, 2021
7018e1a
Fix selection of CPU device
lantiga Mar 14, 2021
95c3a1e
Fix CPU execution
lantiga Mar 14, 2021
865dea9
Set GPU test back
lantiga Mar 14, 2021
22d1b41
Set allow_growth=1
lantiga Mar 14, 2021
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
Prev Previous commit
Next Next commit
Set GPU test back
  • Loading branch information
@lantiga
lantiga committed Mar 14, 2021
commit 865dea9960fdb3f626d9fd66f3c2588beabf88d2
1 change: 0 additions & 1 deletion tests/flow/tests_tensorflow.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -169,7 +169,6 @@ def test_run_tf_model(env):

with open(model_filename, 'rb') as f:
model_pb = f.read()
DEVICE = "CPU"

ret = con.execute_command('AI.MODELSET', 'm{1}', 'TF', DEVICE,
'INPUTS', 'a', 'b', 'OUTPUTS', 'mul', 'BLOB', model_pb)
Expand Down