mirror of
https://github.com/tensorflow/haskell.git
synced 2024-11-26 21:09:44 +01:00
parent
99f8c8cf56
commit
61e58fd33f
22 changed files with 106 additions and 51 deletions
|
@ -16,11 +16,10 @@ packages:
|
||||||
extra-deps:
|
extra-deps:
|
||||||
- snappy-framing-0.1.1
|
- snappy-framing-0.1.1
|
||||||
- snappy-0.2.0.2
|
- snappy-0.2.0.2
|
||||||
# `haskell-src-exts < 1.20` is needed due the dependency on
|
- proto-lens-protobuf-types-0.3.0.1
|
||||||
# `proto-lens-protoc < 0.3` in the `tensorflow-proto` package:
|
- proto-lens-protoc-0.3.1.0
|
||||||
- haskell-src-exts-1.19.1
|
- proto-lens-0.3.1.0
|
||||||
- proto-lens-protobuf-types-0.2.2.0
|
- lens-labels-0.2.0.1
|
||||||
- proto-lens-protoc-0.2.2.3
|
|
||||||
|
|
||||||
# For Mac OS X, whose linker doesn't use this path by default
|
# For Mac OS X, whose linker doesn't use this path by default
|
||||||
# unless you run `xcode-select --install`.
|
# unless you run `xcode-select --install`.
|
||||||
|
|
|
@ -15,7 +15,7 @@ cabal-version: >=1.24
|
||||||
library
|
library
|
||||||
exposed-modules: TensorFlow.GenOps.Core
|
exposed-modules: TensorFlow.GenOps.Core
|
||||||
build-depends: bytestring
|
build-depends: bytestring
|
||||||
, proto-lens == 0.2.*
|
, proto-lens == 0.3.*
|
||||||
, tensorflow == 0.2.*
|
, tensorflow == 0.2.*
|
||||||
, base >= 4.7 && < 5
|
, base >= 4.7 && < 5
|
||||||
, lens-family
|
, lens-family
|
||||||
|
@ -26,7 +26,7 @@ custom-setup
|
||||||
setup-depends: Cabal
|
setup-depends: Cabal
|
||||||
, bytestring
|
, bytestring
|
||||||
, directory
|
, directory
|
||||||
, proto-lens == 0.2.*
|
, proto-lens == 0.3.*
|
||||||
, tensorflow-opgen == 0.2.*
|
, tensorflow-opgen == 0.2.*
|
||||||
, tensorflow == 0.2.*
|
, tensorflow == 0.2.*
|
||||||
, base >= 4.7 && < 5
|
, base >= 4.7 && < 5
|
||||||
|
|
|
@ -69,7 +69,8 @@ import Data.Time.Clock.POSIX (utcTimeToPOSIXSeconds)
|
||||||
import Lens.Family2 ((.~), (&))
|
import Lens.Family2 ((.~), (&))
|
||||||
import Network.HostName (getHostName)
|
import Network.HostName (getHostName)
|
||||||
import Proto.Tensorflow.Core.Framework.Summary (Summary)
|
import Proto.Tensorflow.Core.Framework.Summary (Summary)
|
||||||
import Proto.Tensorflow.Core.Util.Event (Event, fileVersion, graphDef, step, summary, wallTime)
|
import Proto.Tensorflow.Core.Util.Event (Event)
|
||||||
|
import Proto.Tensorflow.Core.Util.Event_Fields (fileVersion, graphDef, step, summary, wallTime)
|
||||||
import System.Directory (createDirectoryIfMissing)
|
import System.Directory (createDirectoryIfMissing)
|
||||||
import System.FilePath ((</>))
|
import System.FilePath ((</>))
|
||||||
import TensorFlow.Build (MonadBuild, Build, asGraphDef)
|
import TensorFlow.Build (MonadBuild, Build, asGraphDef)
|
||||||
|
|
|
@ -24,7 +24,7 @@ library
|
||||||
, filepath
|
, filepath
|
||||||
, hostname
|
, hostname
|
||||||
, lens-family
|
, lens-family
|
||||||
, proto-lens == 0.2.*
|
, proto-lens == 0.3.*
|
||||||
, resourcet
|
, resourcet
|
||||||
, stm
|
, stm
|
||||||
, stm-chans
|
, stm-chans
|
||||||
|
|
|
@ -21,7 +21,8 @@ import Data.Default (def)
|
||||||
import Data.List ((\\))
|
import Data.List ((\\))
|
||||||
import Data.ProtoLens (encodeMessage, decodeMessageOrDie)
|
import Data.ProtoLens (encodeMessage, decodeMessageOrDie)
|
||||||
import Lens.Family2 ((^.), (.~), (&))
|
import Lens.Family2 ((^.), (.~), (&))
|
||||||
import Proto.Tensorflow.Core.Util.Event (Event, graphDef, fileVersion, step)
|
import Proto.Tensorflow.Core.Util.Event (Event)
|
||||||
|
import Proto.Tensorflow.Core.Util.Event_Fields (graphDef, fileVersion, step)
|
||||||
import System.Directory (getDirectoryContents)
|
import System.Directory (getDirectoryContents)
|
||||||
import System.FilePath ((</>))
|
import System.FilePath ((</>))
|
||||||
import System.IO.Temp (withSystemTempDirectory)
|
import System.IO.Temp (withSystemTempDirectory)
|
||||||
|
|
|
@ -20,7 +20,7 @@ library
|
||||||
exposed-modules: TensorFlow.Examples.MNIST.Parse
|
exposed-modules: TensorFlow.Examples.MNIST.Parse
|
||||||
, TensorFlow.Examples.MNIST.TrainedGraph
|
, TensorFlow.Examples.MNIST.TrainedGraph
|
||||||
other-modules: Paths_tensorflow_mnist
|
other-modules: Paths_tensorflow_mnist
|
||||||
build-depends: proto-lens == 0.2.*
|
build-depends: proto-lens == 0.3.*
|
||||||
, base >= 4.7 && < 5
|
, base >= 4.7 && < 5
|
||||||
, binary
|
, binary
|
||||||
, bytestring
|
, bytestring
|
||||||
|
|
|
@ -24,12 +24,13 @@ import qualified Data.Text.IO as Text
|
||||||
import Lens.Family2 ((&), (.~), (^.))
|
import Lens.Family2 ((&), (.~), (^.))
|
||||||
import Prelude hiding (abs)
|
import Prelude hiding (abs)
|
||||||
import Proto.Tensorflow.Core.Framework.Graph
|
import Proto.Tensorflow.Core.Framework.Graph
|
||||||
( GraphDef(..)
|
( GraphDef(..) )
|
||||||
, version
|
import Proto.Tensorflow.Core.Framework.Graph_Fields
|
||||||
|
( version
|
||||||
, node )
|
, node )
|
||||||
import Proto.Tensorflow.Core.Framework.NodeDef
|
import Proto.Tensorflow.Core.Framework.NodeDef
|
||||||
( NodeDef(..)
|
( NodeDef(..) )
|
||||||
, op )
|
import Proto.Tensorflow.Core.Framework.NodeDef_Fields (op)
|
||||||
import System.IO as IO
|
import System.IO as IO
|
||||||
import TensorFlow.Examples.MNIST.InputData
|
import TensorFlow.Examples.MNIST.InputData
|
||||||
import TensorFlow.Examples.MNIST.Parse
|
import TensorFlow.Examples.MNIST.Parse
|
||||||
|
|
|
@ -60,7 +60,9 @@ import Options.Applicative (Parser, help, long, strOption, value)
|
||||||
import Proto.Tensorflow.Core.Framework.OpDef
|
import Proto.Tensorflow.Core.Framework.OpDef
|
||||||
( OpList
|
( OpList
|
||||||
, OpDef
|
, OpDef
|
||||||
, attr
|
)
|
||||||
|
import Proto.Tensorflow.Core.Framework.OpDef_Fields
|
||||||
|
( attr
|
||||||
, inputArg
|
, inputArg
|
||||||
, name
|
, name
|
||||||
, op
|
, op
|
||||||
|
|
|
@ -31,12 +31,14 @@ import qualified Data.Set as Set
|
||||||
import Data.Text (Text)
|
import Data.Text (Text)
|
||||||
import qualified Data.Text as Text
|
import qualified Data.Text as Text
|
||||||
import Lens.Family2 ((^.))
|
import Lens.Family2 ((^.))
|
||||||
import Proto.Tensorflow.Core.Framework.AttrValue (list)
|
import Proto.Tensorflow.Core.Framework.AttrValue_Fields (list)
|
||||||
import Proto.Tensorflow.Core.Framework.OpDef
|
import Proto.Tensorflow.Core.Framework.OpDef
|
||||||
( OpDef
|
( OpDef
|
||||||
, OpDef'ArgDef
|
, OpDef'ArgDef
|
||||||
, OpDef'AttrDef
|
, OpDef'AttrDef
|
||||||
, allowedValues
|
)
|
||||||
|
import Proto.Tensorflow.Core.Framework.OpDef_Fields
|
||||||
|
( allowedValues
|
||||||
, attr
|
, attr
|
||||||
, maybe'defaultValue
|
, maybe'defaultValue
|
||||||
, description
|
, description
|
||||||
|
@ -51,6 +53,7 @@ import Proto.Tensorflow.Core.Framework.OpDef
|
||||||
, typeAttr
|
, typeAttr
|
||||||
, type'
|
, type'
|
||||||
)
|
)
|
||||||
|
|
||||||
import Proto.Tensorflow.Core.Framework.Types (DataType(DT_RESOURCE))
|
import Proto.Tensorflow.Core.Framework.Types (DataType(DT_RESOURCE))
|
||||||
|
|
||||||
data ParsedOp = ParsedOp
|
data ParsedOp = ParsedOp
|
||||||
|
|
|
@ -16,7 +16,7 @@ library
|
||||||
hs-source-dirs: src
|
hs-source-dirs: src
|
||||||
exposed-modules: TensorFlow.OpGen.ParsedOp
|
exposed-modules: TensorFlow.OpGen.ParsedOp
|
||||||
, TensorFlow.OpGen
|
, TensorFlow.OpGen
|
||||||
build-depends: proto-lens == 0.2.*
|
build-depends: proto-lens == 0.3.*
|
||||||
, tensorflow-proto == 0.2.*
|
, tensorflow-proto == 0.2.*
|
||||||
, base >= 4.7 && < 5
|
, base >= 4.7 && < 5
|
||||||
, bytestring
|
, bytestring
|
||||||
|
|
|
@ -103,8 +103,9 @@ import TensorFlow.Tensor
|
||||||
, ToTensor(..)
|
, ToTensor(..)
|
||||||
)
|
)
|
||||||
import TensorFlow.Types (Attribute, OneOf, TensorType, attrLens)
|
import TensorFlow.Types (Attribute, OneOf, TensorType, attrLens)
|
||||||
import Proto.Tensorflow.Core.Framework.NodeDef
|
import Proto.Tensorflow.Core.Framework.NodeDef (NodeDef)
|
||||||
(NodeDef, attr, input, op, name)
|
import Proto.Tensorflow.Core.Framework.NodeDef_Fields
|
||||||
|
( attr, input, op, name)
|
||||||
|
|
||||||
type GradientCompatible a =
|
type GradientCompatible a =
|
||||||
-- TODO(fmayle): MaxPoolGrad doesn't support Double for some reason.
|
-- TODO(fmayle): MaxPoolGrad doesn't support Double for some reason.
|
||||||
|
|
|
@ -160,13 +160,14 @@ import Data.ProtoLens (def)
|
||||||
import Data.Text.Encoding (encodeUtf8)
|
import Data.Text.Encoding (encodeUtf8)
|
||||||
import Lens.Family2 ((.~), (&))
|
import Lens.Family2 ((.~), (&))
|
||||||
import Text.Printf (printf)
|
import Text.Printf (printf)
|
||||||
import Proto.Tensorflow.Core.Framework.Tensor
|
import Proto.Tensorflow.Core.Framework.Tensor (TensorProto)
|
||||||
( TensorProto
|
import Proto.Tensorflow.Core.Framework.Tensor_Fields
|
||||||
, dtype
|
( dtype
|
||||||
, tensorShape
|
, tensorShape
|
||||||
)
|
)
|
||||||
import qualified Proto.Tensorflow.Core.Framework.TensorShape
|
import qualified Proto.Tensorflow.Core.Framework.TensorShape_Fields
|
||||||
as TensorShape
|
as TensorShape
|
||||||
|
|
||||||
import TensorFlow.Build
|
import TensorFlow.Build
|
||||||
import TensorFlow.BuildOp
|
import TensorFlow.BuildOp
|
||||||
import TensorFlow.ControlFlow (group)
|
import TensorFlow.ControlFlow (group)
|
||||||
|
|
|
@ -21,7 +21,7 @@ library
|
||||||
, TensorFlow.NN
|
, TensorFlow.NN
|
||||||
, TensorFlow.Queue
|
, TensorFlow.Queue
|
||||||
, TensorFlow.Variable
|
, TensorFlow.Variable
|
||||||
build-depends: proto-lens == 0.2.*
|
build-depends: proto-lens == 0.3.*
|
||||||
, base >= 4.7 && < 5
|
, base >= 4.7 && < 5
|
||||||
, bytestring
|
, bytestring
|
||||||
, fgl
|
, fgl
|
||||||
|
|
|
@ -21,11 +21,12 @@ module Main where
|
||||||
import Control.Monad.IO.Class (liftIO)
|
import Control.Monad.IO.Class (liftIO)
|
||||||
import Lens.Family2 ((^.), (.~))
|
import Lens.Family2 ((^.), (.~))
|
||||||
import Data.List (sort)
|
import Data.List (sort)
|
||||||
import Proto.Tensorflow.Core.Framework.Graph
|
import Proto.Tensorflow.Core.Framework.Graph_Fields
|
||||||
( node )
|
( node )
|
||||||
import Proto.Tensorflow.Core.Framework.NodeDef
|
import Proto.Tensorflow.Core.Framework.NodeDef
|
||||||
( NodeDef
|
( NodeDef )
|
||||||
, device
|
import Proto.Tensorflow.Core.Framework.NodeDef_Fields
|
||||||
|
( device
|
||||||
, name
|
, name
|
||||||
, op )
|
, op )
|
||||||
import TensorFlow.Build
|
import TensorFlow.Build
|
||||||
|
|
|
@ -39,8 +39,8 @@ import qualified TensorFlow.Output as TF
|
||||||
import qualified TensorFlow.Types as TF
|
import qualified TensorFlow.Types as TF
|
||||||
import qualified TensorFlow.Variable as TF
|
import qualified TensorFlow.Variable as TF
|
||||||
|
|
||||||
import Proto.Tensorflow.Core.Framework.Graph (node)
|
import Proto.Tensorflow.Core.Framework.Graph_Fields (node)
|
||||||
import Proto.Tensorflow.Core.Framework.NodeDef (op)
|
import Proto.Tensorflow.Core.Framework.NodeDef_Fields (op)
|
||||||
|
|
||||||
import qualified Data.ByteString.Char8 as BS
|
import qualified Data.ByteString.Char8 as BS
|
||||||
|
|
||||||
|
|
|
@ -19,55 +19,94 @@ extra-source-files: third_party/tensorflow/tensorflow/core/example/*.proto
|
||||||
|
|
||||||
library
|
library
|
||||||
exposed-modules: Proto.Tensorflow.Core.Example.Example
|
exposed-modules: Proto.Tensorflow.Core.Example.Example
|
||||||
|
, Proto.Tensorflow.Core.Example.Example_Fields
|
||||||
, Proto.Tensorflow.Core.Example.ExampleParserConfiguration
|
, Proto.Tensorflow.Core.Example.ExampleParserConfiguration
|
||||||
|
, Proto.Tensorflow.Core.Example.ExampleParserConfiguration_Fields
|
||||||
, Proto.Tensorflow.Core.Example.Feature
|
, Proto.Tensorflow.Core.Example.Feature
|
||||||
|
, Proto.Tensorflow.Core.Example.Feature_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.AllocationDescription
|
, Proto.Tensorflow.Core.Framework.AllocationDescription
|
||||||
|
, Proto.Tensorflow.Core.Framework.AllocationDescription_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.AttrValue
|
, Proto.Tensorflow.Core.Framework.AttrValue
|
||||||
|
, Proto.Tensorflow.Core.Framework.AttrValue_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.CostGraph
|
, Proto.Tensorflow.Core.Framework.CostGraph
|
||||||
|
, Proto.Tensorflow.Core.Framework.CostGraph_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.DeviceAttributes
|
, Proto.Tensorflow.Core.Framework.DeviceAttributes
|
||||||
|
, Proto.Tensorflow.Core.Framework.DeviceAttributes_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.Function
|
, Proto.Tensorflow.Core.Framework.Function
|
||||||
|
, Proto.Tensorflow.Core.Framework.Function_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.Graph
|
, Proto.Tensorflow.Core.Framework.Graph
|
||||||
|
, Proto.Tensorflow.Core.Framework.Graph_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.KernelDef
|
, Proto.Tensorflow.Core.Framework.KernelDef
|
||||||
|
, Proto.Tensorflow.Core.Framework.KernelDef_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.LogMemory
|
, Proto.Tensorflow.Core.Framework.LogMemory
|
||||||
|
, Proto.Tensorflow.Core.Framework.LogMemory_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.NodeDef
|
, Proto.Tensorflow.Core.Framework.NodeDef
|
||||||
|
, Proto.Tensorflow.Core.Framework.NodeDef_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.OpDef
|
, Proto.Tensorflow.Core.Framework.OpDef
|
||||||
|
, Proto.Tensorflow.Core.Framework.OpDef_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.ResourceHandle
|
, Proto.Tensorflow.Core.Framework.ResourceHandle
|
||||||
|
, Proto.Tensorflow.Core.Framework.ResourceHandle_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.StepStats
|
, Proto.Tensorflow.Core.Framework.StepStats
|
||||||
|
, Proto.Tensorflow.Core.Framework.StepStats_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.Summary
|
, Proto.Tensorflow.Core.Framework.Summary
|
||||||
|
, Proto.Tensorflow.Core.Framework.Summary_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.Tensor
|
, Proto.Tensorflow.Core.Framework.Tensor
|
||||||
|
, Proto.Tensorflow.Core.Framework.Tensor_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.TensorDescription
|
, Proto.Tensorflow.Core.Framework.TensorDescription
|
||||||
|
, Proto.Tensorflow.Core.Framework.TensorDescription_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.TensorShape
|
, Proto.Tensorflow.Core.Framework.TensorShape
|
||||||
|
, Proto.Tensorflow.Core.Framework.TensorShape_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.TensorSlice
|
, Proto.Tensorflow.Core.Framework.TensorSlice
|
||||||
|
, Proto.Tensorflow.Core.Framework.TensorSlice_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.Types
|
, Proto.Tensorflow.Core.Framework.Types
|
||||||
|
, Proto.Tensorflow.Core.Framework.Types_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.Variable
|
, Proto.Tensorflow.Core.Framework.Variable
|
||||||
|
, Proto.Tensorflow.Core.Framework.Variable_Fields
|
||||||
, Proto.Tensorflow.Core.Framework.Versions
|
, Proto.Tensorflow.Core.Framework.Versions
|
||||||
|
, Proto.Tensorflow.Core.Framework.Versions_Fields
|
||||||
, Proto.Tensorflow.Core.Lib.Core.ErrorCodes
|
, Proto.Tensorflow.Core.Lib.Core.ErrorCodes
|
||||||
, Proto.Tensorflow.Core.Protobuf.Cluster
|
, Proto.Tensorflow.Core.Protobuf.Cluster
|
||||||
|
, Proto.Tensorflow.Core.Protobuf.Cluster_Fields
|
||||||
, Proto.Tensorflow.Core.Protobuf.Config
|
, Proto.Tensorflow.Core.Protobuf.Config
|
||||||
|
, Proto.Tensorflow.Core.Protobuf.Config_Fields
|
||||||
, Proto.Tensorflow.Core.Protobuf.ControlFlow
|
, Proto.Tensorflow.Core.Protobuf.ControlFlow
|
||||||
|
, Proto.Tensorflow.Core.Protobuf.ControlFlow_Fields
|
||||||
, Proto.Tensorflow.Core.Protobuf.Debug
|
, Proto.Tensorflow.Core.Protobuf.Debug
|
||||||
|
, Proto.Tensorflow.Core.Protobuf.Debug_Fields
|
||||||
, Proto.Tensorflow.Core.Protobuf.MetaGraph
|
, Proto.Tensorflow.Core.Protobuf.MetaGraph
|
||||||
|
, Proto.Tensorflow.Core.Protobuf.MetaGraph_Fields
|
||||||
, Proto.Tensorflow.Core.Protobuf.NamedTensor
|
, Proto.Tensorflow.Core.Protobuf.NamedTensor
|
||||||
|
, Proto.Tensorflow.Core.Protobuf.NamedTensor_Fields
|
||||||
, Proto.Tensorflow.Core.Protobuf.QueueRunner
|
, Proto.Tensorflow.Core.Protobuf.QueueRunner
|
||||||
|
, Proto.Tensorflow.Core.Protobuf.QueueRunner_Fields
|
||||||
, Proto.Tensorflow.Core.Protobuf.RewriterConfig
|
, Proto.Tensorflow.Core.Protobuf.RewriterConfig
|
||||||
|
, Proto.Tensorflow.Core.Protobuf.RewriterConfig_Fields
|
||||||
, Proto.Tensorflow.Core.Protobuf.SavedModel
|
, Proto.Tensorflow.Core.Protobuf.SavedModel
|
||||||
|
, Proto.Tensorflow.Core.Protobuf.SavedModel_Fields
|
||||||
, Proto.Tensorflow.Core.Protobuf.Saver
|
, Proto.Tensorflow.Core.Protobuf.Saver
|
||||||
|
, Proto.Tensorflow.Core.Protobuf.Saver_Fields
|
||||||
, Proto.Tensorflow.Core.Protobuf.TensorBundle
|
, Proto.Tensorflow.Core.Protobuf.TensorBundle
|
||||||
|
, Proto.Tensorflow.Core.Protobuf.TensorBundle_Fields
|
||||||
, Proto.Tensorflow.Core.Protobuf.TensorflowServer
|
, Proto.Tensorflow.Core.Protobuf.TensorflowServer
|
||||||
|
, Proto.Tensorflow.Core.Protobuf.TensorflowServer_Fields
|
||||||
, Proto.Tensorflow.Core.Util.Event
|
, Proto.Tensorflow.Core.Util.Event
|
||||||
|
, Proto.Tensorflow.Core.Util.Event_Fields
|
||||||
, Proto.Tensorflow.Core.Util.MemmappedFileSystem
|
, Proto.Tensorflow.Core.Util.MemmappedFileSystem
|
||||||
|
, Proto.Tensorflow.Core.Util.MemmappedFileSystem_Fields
|
||||||
, Proto.Tensorflow.Core.Util.SavedTensorSlice
|
, Proto.Tensorflow.Core.Util.SavedTensorSlice
|
||||||
|
, Proto.Tensorflow.Core.Util.SavedTensorSlice_Fields
|
||||||
, Proto.Tensorflow.Core.Util.TestLog
|
, Proto.Tensorflow.Core.Util.TestLog
|
||||||
build-depends: proto-lens >= 0.2.2 && < 0.3
|
, Proto.Tensorflow.Core.Util.TestLog_Fields
|
||||||
, proto-lens-protoc >= 0.2.2.1 && < 0.3
|
build-depends: proto-lens == 0.3.*
|
||||||
, proto-lens-protobuf-types == 0.2.*
|
, proto-lens-protoc == 0.3.*
|
||||||
|
, proto-lens-protobuf-types == 0.3.*
|
||||||
, base >= 4.7 && < 5
|
, base >= 4.7 && < 5
|
||||||
default-language: Haskell2010
|
default-language: Haskell2010
|
||||||
include-dirs: .
|
include-dirs: .
|
||||||
|
|
||||||
custom-setup
|
custom-setup
|
||||||
setup-depends: Cabal
|
setup-depends: Cabal
|
||||||
, proto-lens-protoc == 0.2.*
|
, proto-lens-protoc == 0.3.*
|
||||||
, base >= 4.7 && < 5
|
, base >= 4.7 && < 5
|
||||||
source-repository head
|
source-repository head
|
||||||
type: git
|
type: git
|
||||||
|
|
|
@ -78,13 +78,11 @@ import qualified Data.Text as Text
|
||||||
import Lens.Family2 (Lens', (.~), (^.), (&))
|
import Lens.Family2 (Lens', (.~), (^.), (&))
|
||||||
import Lens.Family2.State.Strict (MonadState, use, uses, (.=), (<>=), (%=))
|
import Lens.Family2.State.Strict (MonadState, use, uses, (.=), (<>=), (%=))
|
||||||
import Lens.Family2.Unchecked (lens)
|
import Lens.Family2.Unchecked (lens)
|
||||||
import Proto.Tensorflow.Core.Framework.Graph
|
import Proto.Tensorflow.Core.Framework.Graph (GraphDef)
|
||||||
( GraphDef
|
import Proto.Tensorflow.Core.Framework.Graph_Fields (node)
|
||||||
, node
|
import Proto.Tensorflow.Core.Framework.NodeDef (NodeDef)
|
||||||
)
|
import Proto.Tensorflow.Core.Framework.NodeDef_Fields
|
||||||
import Proto.Tensorflow.Core.Framework.NodeDef
|
( attr
|
||||||
( NodeDef
|
|
||||||
, attr
|
|
||||||
, input
|
, input
|
||||||
, device
|
, device
|
||||||
, name
|
, name
|
||||||
|
|
|
@ -50,7 +50,8 @@ import Data.Set (Set)
|
||||||
import Data.Text.Encoding (encodeUtf8)
|
import Data.Text.Encoding (encodeUtf8)
|
||||||
import Lens.Family2 (Lens', (^.), (&), (.~))
|
import Lens.Family2 (Lens', (^.), (&), (.~))
|
||||||
import Lens.Family2.Unchecked (lens)
|
import Lens.Family2.Unchecked (lens)
|
||||||
import Proto.Tensorflow.Core.Framework.Graph (GraphDef, node)
|
import Proto.Tensorflow.Core.Framework.Graph (GraphDef)
|
||||||
|
import Proto.Tensorflow.Core.Framework.Graph_Fields (node)
|
||||||
import Proto.Tensorflow.Core.Protobuf.Config (ConfigProto)
|
import Proto.Tensorflow.Core.Protobuf.Config (ConfigProto)
|
||||||
import TensorFlow.Build
|
import TensorFlow.Build
|
||||||
import TensorFlow.Nodes
|
import TensorFlow.Nodes
|
||||||
|
|
|
@ -34,7 +34,7 @@ import qualified Data.Text as Text
|
||||||
import Lens.Family2 ((^.))
|
import Lens.Family2 ((^.))
|
||||||
import Lens.Family2.State ((%=), use)
|
import Lens.Family2.State ((%=), use)
|
||||||
|
|
||||||
import Proto.Tensorflow.Core.Framework.NodeDef (device)
|
import Proto.Tensorflow.Core.Framework.NodeDef_Fields (device)
|
||||||
import TensorFlow.Build
|
import TensorFlow.Build
|
||||||
import TensorFlow.Output (Output, NodeName, outputNodeName, Device(..))
|
import TensorFlow.Output (Output, NodeName, outputNodeName, Device(..))
|
||||||
import TensorFlow.Types
|
import TensorFlow.Types
|
||||||
|
|
|
@ -90,7 +90,9 @@ import qualified Data.Vector.Storable as S
|
||||||
import Proto.Tensorflow.Core.Framework.AttrValue
|
import Proto.Tensorflow.Core.Framework.AttrValue
|
||||||
( AttrValue(..)
|
( AttrValue(..)
|
||||||
, AttrValue'ListValue(..)
|
, AttrValue'ListValue(..)
|
||||||
, b
|
)
|
||||||
|
import Proto.Tensorflow.Core.Framework.AttrValue_Fields
|
||||||
|
( b
|
||||||
, f
|
, f
|
||||||
, i
|
, i
|
||||||
, s
|
, s
|
||||||
|
@ -99,11 +101,13 @@ import Proto.Tensorflow.Core.Framework.AttrValue
|
||||||
, shape
|
, shape
|
||||||
, tensor
|
, tensor
|
||||||
)
|
)
|
||||||
|
|
||||||
import Proto.Tensorflow.Core.Framework.ResourceHandle
|
import Proto.Tensorflow.Core.Framework.ResourceHandle
|
||||||
(ResourceHandleProto)
|
(ResourceHandleProto)
|
||||||
import Proto.Tensorflow.Core.Framework.Tensor as Tensor
|
import Proto.Tensorflow.Core.Framework.Tensor as Tensor
|
||||||
( TensorProto(..)
|
(TensorProto(..))
|
||||||
, boolVal
|
import Proto.Tensorflow.Core.Framework.Tensor_Fields as Tensor
|
||||||
|
( boolVal
|
||||||
, doubleVal
|
, doubleVal
|
||||||
, floatVal
|
, floatVal
|
||||||
, intVal
|
, intVal
|
||||||
|
@ -113,9 +117,11 @@ import Proto.Tensorflow.Core.Framework.Tensor as Tensor
|
||||||
, uint32Val
|
, uint32Val
|
||||||
, uint64Val
|
, uint64Val
|
||||||
)
|
)
|
||||||
|
|
||||||
import Proto.Tensorflow.Core.Framework.TensorShape
|
import Proto.Tensorflow.Core.Framework.TensorShape
|
||||||
( TensorShapeProto(..)
|
(TensorShapeProto(..))
|
||||||
, dim
|
import Proto.Tensorflow.Core.Framework.TensorShape_Fields
|
||||||
|
( dim
|
||||||
, size
|
, size
|
||||||
, unknownRank
|
, unknownRank
|
||||||
)
|
)
|
||||||
|
|
|
@ -36,7 +36,7 @@ library
|
||||||
, TensorFlow.Types
|
, TensorFlow.Types
|
||||||
other-modules: TensorFlow.Internal.Raw
|
other-modules: TensorFlow.Internal.Raw
|
||||||
build-tools: c2hs
|
build-tools: c2hs
|
||||||
build-depends: proto-lens == 0.2.*
|
build-depends: proto-lens == 0.3.*
|
||||||
, tensorflow-proto == 0.2.*
|
, tensorflow-proto == 0.2.*
|
||||||
, base >= 4.7 && < 5
|
, base >= 4.7 && < 5
|
||||||
, async
|
, async
|
||||||
|
|
|
@ -22,7 +22,8 @@ import TensorFlow.Internal.FFI (getAllOpList)
|
||||||
import Test.HUnit (assertBool, assertFailure)
|
import Test.HUnit (assertBool, assertFailure)
|
||||||
import Test.Framework (defaultMain)
|
import Test.Framework (defaultMain)
|
||||||
import Test.Framework.Providers.HUnit (testCase)
|
import Test.Framework.Providers.HUnit (testCase)
|
||||||
import Proto.Tensorflow.Core.Framework.OpDef (OpList, op)
|
import Proto.Tensorflow.Core.Framework.OpDef (OpList)
|
||||||
|
import Proto.Tensorflow.Core.Framework.OpDef_Fields (op)
|
||||||
|
|
||||||
testParseAll :: IO ()
|
testParseAll :: IO ()
|
||||||
testParseAll = do
|
testParseAll = do
|
||||||
|
|
Loading…
Reference in a new issue