Skip to content

Conversation

pschuh
Copy link
Contributor

@pschuh pschuh commented Apr 25, 2019

Two problems:

  • When returning a list of tensors, the CTensorHandles represent a reference. These need to be passed owning.
  • TFE_Ops will leak tensors if not freed.

@pschuh pschuh requested a review from bgogul April 25, 2019 23:57
@bgogul
Copy link
Contributor

bgogul commented Apr 25, 2019

Nice!

@pschuh
Copy link
Contributor Author

pschuh commented Apr 26, 2019

@swift-ci please test TensorFlow

@pschuh
Copy link
Contributor Author

pschuh commented Apr 26, 2019

@swift-ci please test TensorFlow gpu

Copy link
Contributor

@rxwei rxwei left a comment

Choose a reason for hiding this comment

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

yay!

self.init(_owning: buffer)
}

init<C: Collection>(_owning input: C) where C.Element == CTensorHandle {
Copy link
Contributor

@rxwei rxwei Apr 26, 2019

Choose a reason for hiding this comment

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

Suggested change
init<C: Collection>(_owning input: C) where C.Element == CTensorHandle {
init<C : Collection>(_owning input: C) where C.Element == CTensorHandle {

(Random nit. No big deal.)

@pschuh
Copy link
Contributor Author

pschuh commented Apr 26, 2019

@swift-ci please test TensorFlow gpu

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

pschuh commented Apr 26, 2019

GPU failure seems unrelated (tensor slicing)

@pschuh pschuh merged commit 57f8cf5 into swiftlang:tensorflow Apr 26, 2019
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.

4 participants