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

Fix compilation with tensorflow 2.12 and higher

This commit is contained in:
Jeroen Bransen 2023-08-10 13:43:53 +02:00
parent bfd8de5582
commit 7166edb29c
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -261,7 +261,7 @@ resolveOperation graph name = do
where
exception =
let msg = "Operation not found in graph: " <> (T.pack $ show name)
in TensorFlowException Raw.TF_INVALID_ARGUMENT msg
in TensorFlowException Raw.TSL_INVALID_ARGUMENT msg
-- Internal.
@ -325,7 +325,7 @@ checkStatus fn =
bracket Raw.newStatus Raw.deleteStatus $ \status -> do
result <- fn status
code <- Raw.getCode status
when (code /= Raw.TF_OK) $ do
when (code /= Raw.TSL_OK) $ do
msg <- T.decodeUtf8With T.lenientDecode <$>
(Raw.message status >>= B.packCString)
throwM $ TensorFlowException code msg
@ -372,4 +372,4 @@ getAllOpList =
when (p == nullPtr) (throwM exception)
return p
exception = TensorFlowException
Raw.TF_UNKNOWN "GetAllOpList failure, check logs"
Raw.TSL_UNKNOWN "GetAllOpList failure, check logs"