eeva/tensorflow-haskell
1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-12-24 18:49:46 +01:00
Code Issues Releases Wiki Activity

Sort the ops generated in TensorFlow.GenOps.Core. (#96)

Browse source
This commit is contained in:
Judah Jacobson 2017-04-08 07:15:28 -07:00 committed by fkm3
parent 55ca545915
commit de16a576da
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tensorflow-opgen/src/TensorFlow/OpGen.hs
View file

@ -53,6 +53,7 @@ module TensorFlow.OpGen
import Data.Foldable (toList)
import Data.Maybe (fromMaybe)
import Data.ProtoLens (def, showMessage)
import Data.List (sortOn)
import Data.List.NonEmpty (NonEmpty)
import qualified Data.List.NonEmpty as NE
import Lens.Family2 ((^.), (.~), (&), view)
@ -132,6 +133,7 @@ docOpList flags opList =
, empty
, folddoc (\x y -> x </> empty </> y)
(map renderOpAndExtras $
sortOn (view name) $
filter (not . flip elem exclusions . view name) $
toList $ opList ^. op)
]