eeva/tensorflow-haskell
1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-11-26 21:09:44 +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.Foldable (toList)
import Data.Maybe (fromMaybe) import Data.Maybe (fromMaybe)
import Data.ProtoLens (def, showMessage) import Data.ProtoLens (def, showMessage)
import Data.List (sortOn)
import Data.List.NonEmpty (NonEmpty) import Data.List.NonEmpty (NonEmpty)
import qualified Data.List.NonEmpty as NE import qualified Data.List.NonEmpty as NE
import Lens.Family2 ((^.), (.~), (&), view) import Lens.Family2 ((^.), (.~), (&), view)
@ -132,6 +133,7 @@ docOpList flags opList =
, empty , empty
, folddoc (\x y -> x </> empty </> y) , folddoc (\x y -> x </> empty </> y)
(map renderOpAndExtras $ (map renderOpAndExtras $
sortOn (view name) $
filter (not . flip elem exclusions . view name) $ filter (not . flip elem exclusions . view name) $
toList $ opList ^. op) toList $ opList ^. op)
] ]