eeva/tensorflow-haskell
1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-11-23 03:19:44 +01:00
Code Issues Releases Wiki Activity

Fix a small typo in the warning message of numOutputs (#151)

Browse source
This commit is contained in:
Nathan T.A. Lewis 2017-08-24 15:34:22 -06:00 committed by fkm3
parent 7e0ef4bcd3
commit d79a919efa
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tensorflow-ops/src/TensorFlow/Gradient.hs
View file

@ -817,7 +817,7 @@ numOutputs o =
"Variable" -> 1
"ZerosLike" -> 1
"Fill" -> 1
_ -> error $ "numOuputs not implemented for " ++ show (o ^. op)
_ -> error $ "numOutputs not implemented for " ++ show (o ^. op)
-- Divides `x / y` assuming `x, y >= 0`, treating `0 / 0 = 0`
safeShapeDiv :: Tensor v1 Int32 -> Tensor v2 Int32 -> Tensor Build Int32