Skip to content

Internal wrapper for direct eager dispatching. #23777

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

Merged
merged 1 commit into from
Apr 17, 2019
Merged

Conversation

pschuh
Copy link
Contributor

@pschuh pschuh commented Apr 3, 2019

  • Add support for dispatching an eager op which returns a tensor group.
  • AnyTensor type for array output and input lists.
  • Add proper eager support for saveV2 and restoreV2 ops. These will
    be used for model checkpointing.

@pschuh pschuh requested review from marcrasi and bgogul April 3, 2019 21:04
internal func makeTensor(owning pointer: CTensorHandle) -> AnyTensor {
switch (_cDataType) {
case TF_BOOL:
return Tensor<Bool>.init(handle: TensorHandle(_owning: pointer))
Copy link
Contributor

Choose a reason for hiding this comment

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

Do you need .init?

@pschuh
Copy link
Contributor Author

pschuh commented Apr 3, 2019

@swift-ci please test tensorflow

@rxwei
Copy link
Contributor

rxwei commented Apr 4, 2019

Just a friendly reminder that addition or changes to public APIs without underscore prefixes should go through reviews by API owners. It’s mainly about the newly added public protocol AnyTensor.

@dan-zheng dan-zheng added the tensorflow This is for "tensorflow" branch PRs. label Apr 4, 2019
@pschuh
Copy link
Contributor Author

pschuh commented Apr 4, 2019

@swift-ci please test tensorflow

3 similar comments
@pschuh
Copy link
Contributor Author

pschuh commented Apr 4, 2019

@swift-ci please test tensorflow

@pschuh
Copy link
Contributor Author

pschuh commented Apr 4, 2019

@swift-ci please test tensorflow

@pschuh
Copy link
Contributor Author

pschuh commented Apr 4, 2019

@swift-ci please test tensorflow

@pschuh
Copy link
Contributor Author

pschuh commented Apr 4, 2019

@swift-ci please test tensorflow

- Add support for dispatching an eager op which returns a tensor group.
- AnyTensor type for array output and input lists.
- Add proper eager support for saveV2 and restoreV2 ops. These will
be used for model checkpointing.
@pschuh
Copy link
Contributor Author

pschuh commented Apr 12, 2019

@swift-ci please test tensorflow

@pschuh
Copy link
Contributor Author

pschuh commented Apr 15, 2019

@swift-ci please test macos

@pschuh
Copy link
Contributor Author

pschuh commented Apr 15, 2019

@swift-ci please test tensorflow macos

@pschuh
Copy link
Contributor Author

pschuh commented Apr 15, 2019

@swift-ci please test gpu macos

@pschuh
Copy link
Contributor Author

pschuh commented Apr 15, 2019

@swift-ci please test gpu tensorflow

@pschuh
Copy link
Contributor Author

pschuh commented Apr 15, 2019

@swift-ci please test tensorflow gpu

@swift-ci
Copy link
Contributor

Build failed
Swift Test OS X Platform
Git Sha - 8b500d2

@pschuh
Copy link
Contributor Author

pschuh commented Apr 15, 2019

@swift-ci please test tensorflow macos

@pschuh
Copy link
Contributor Author

pschuh commented Apr 17, 2019

@swift-ci please test tensorflow gpu

@pschuh pschuh merged commit 1fdbae8 into swiftlang:tensorflow Apr 17, 2019
rxwei pushed a commit to rxwei/swift that referenced this pull request May 11, 2019
- Add support for dispatching an eager op which returns a tensor group.
- AnyTensor type for array output and input lists.
- Add proper eager support for saveV2 and restoreV2 ops. These will
be used for model checkpointing.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
tensorflow This is for "tensorflow" branch PRs.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants