-
Notifications
You must be signed in to change notification settings - Fork 10.5k
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
Conversation
internal func makeTensor(owning pointer: CTensorHandle) -> AnyTensor { | ||
switch (_cDataType) { | ||
case TF_BOOL: | ||
return Tensor<Bool>.init(handle: TensorHandle(_owning: pointer)) |
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.
Do you need .init
?
@swift-ci please test tensorflow |
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. |
@swift-ci please test tensorflow |
3 similar comments
@swift-ci please test tensorflow |
@swift-ci please test tensorflow |
@swift-ci please test tensorflow |
@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.
@swift-ci please test tensorflow |
@swift-ci please test macos |
@swift-ci please test tensorflow macos |
@swift-ci please test gpu macos |
@swift-ci please test gpu tensorflow |
@swift-ci please test tensorflow gpu |
Build failed |
@swift-ci please test tensorflow macos |
@swift-ci please test tensorflow gpu |
- 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.
be used for model checkpointing.