eeva
/
tensorflow-haskell
mirror of https://github.com/tensorflow/haskell.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Uprev tensorflow. (#33) 

* No longer need to hide ResourceHandle ops
* Blacklisted not supported TensorArrayV2
* Ownership of feed tensors changed (1f0c5119a0230c5160d45496175b9256f097e144)
This commit is contained in:
Greg Steuck 2016-11-16 21:16:20 -08:00 committed by GitHub
parent ea9ac9e37c
commit c430e54c3c
3 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow-core-ops/Setup.hs
View File

@ -75,6 +75,7 @@ blackList =
, "QueueDequeueUpTo" , "QueueDequeueUpTo"
, "Stack" , "Stack"
, "TensorArray" , "TensorArray"
, "TensorArrayV2"
-- These should be possible to support by adding a bunch of -- These should be possible to support by adding a bunch of
-- overloads with a variable number of tuple arguments. -- overloads with a variable number of tuple arguments.
, "Assert" , "Assert"
@ -99,7 +100,4 @@ blackList =
, "_ListToArray" , "_ListToArray"
-- Easy: support larger result tuples. -- Easy: support larger result tuples.
, "Skipgram" , "Skipgram"
-- Can be removed after 139136489 is released.
, "ResourceGather"
, "ResourceScatterAdd"
] ]

1
tensorflow/src/TensorFlow/Internal/FFI.hs
View File

@ -137,6 +137,7 @@ run session feeds fetches targets = do
fetchNames tensorOuts (fromIntegral fetchesLen) fetchNames tensorOuts (fromIntegral fetchesLen)
ctargets (fromIntegral targetsLen) ctargets (fromIntegral targetsLen)
nullPtr nullPtr
mapM_ Raw.deleteTensor feedTensors
outTensors <- peekArray fetchesLen tensorOuts outTensors <- peekArray fetchesLen tensorOuts
mapM createTensorData outTensors mapM createTensorData outTensors

2
third_party/tensorflow vendored

@ -1 +1 @@
Subproject commit 86f5ab7474825da756838b34e1b4eac93f5fc68a Subproject commit 3d41cf77d624aeee0482f92121a9300b29db2809