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

implement some existed operators for TF #55

Merged
merged 4 commits into from
Sep 26, 2022
Merged

implement some existed operators for TF #55

merged 4 commits into from
Sep 26, 2022

Conversation

Co1lin
Copy link
Collaborator

@Co1lin Co1lin commented Sep 26, 2022

  • implement some existed operators for TF
  • fix the GPU device name for TF (if we directly use GPU device name got by list_physical_devices in TF, we will encounter an exception when using this device via with: ValueError: Unknown attribute 'physical_device' is encountered while parsing the device spec: '/physical_device:GPU:0'.)

Copy link
Member

@ganler ganler left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for the contribution! Just need to confirm with the review comments. Meanwhile, please confirm that dispatchable operators can at least work with one backend, thanks!

nnsmith/materialize/tensorflow/__init__.py Outdated Show resolved Hide resolved
nnsmith/materialize/tensorflow/forward.py Show resolved Hide resolved
nnsmith/materialize/tensorflow/forward.py Outdated Show resolved Hide resolved
nnsmith/materialize/tensorflow/forward.py Outdated Show resolved Hide resolved
@ganler ganler merged commit 1fc1c1c into ise-uiuc:main Sep 26, 2022
@ganler ganler mentioned this pull request Feb 19, 2023
1 task
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants