1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-11-15 07:29:43 +01:00
tensorflow-haskell/docs/haddock/tensorflow-logging-0.2.0.1/tensorflow-logging.txt

68 lines
2.5 KiB
Text
Raw Normal View History

2018-08-14 06:05:49 +02:00
-- Hoogle documentation, generated by Haddock
-- See Hoogle, http://www.haskell.org/hoogle/
-- | TensorBoard related functionality.
--
-- Please see README.md
@package tensorflow-logging
@version 0.2.0.1
2018-08-14 06:05:49 +02:00
-- | TensorBoard Summary generation. Provides type safe wrappers around raw
-- string emitting CoreOps.
--
-- Example use:
--
-- <pre>
-- -- Call summary functions while constructing the graph.
-- createModel = do
-- loss &lt;- -- ...
-- TF.scalarSummary loss
--
-- -- Write summaries to an EventWriter.
-- train = TF.withEventWriter "/path/to/logs" $ \eventWriter -&gt; do
-- summaryTensor &lt;- TF.build TF.allSummaries
-- forM_ [1..] $ \step -&gt; do
-- if (step % 100 == 0)
-- then do
-- ((), summaryBytes) &lt;- TF.run (trainStep, summaryTensor)
-- let summary = decodeMessageOrDie (TF.unScalar summaryBytes)
-- TF.logSummary eventWriter step summary
-- else TF.run_ trainStep
-- </pre>
module TensorFlow.Logging
-- | Handle for logging TensorBoard events safely from multiple threads.
data EventWriter
-- | Writes Event protocol buffers to event files.
withEventWriter :: (MonadIO m, MonadMask m) => FilePath -> (EventWriter -> m a) -> m a
-- | Logs the given Event protocol buffer.
logEvent :: MonadIO m => EventWriter -> Event -> m ()
-- | Logs the graph for the given <a>Build</a> action.
logGraph :: MonadIO m => EventWriter -> Build a -> m ()
-- | Logs the given Summary event with an optional global step (use 0 if
-- not applicable).
logSummary :: MonadIO m => EventWriter -> Int64 -> Summary -> m ()
-- | Synonym for the tensors that return serialized Summary proto.
type SummaryTensor = Tensor Value ByteString
-- | Adds a <a>histogramSummary</a> node. The tag argument is intentionally
-- limited to a single value for simplicity.
histogramSummary :: (MonadBuild m, TensorType t, t /= ByteString, t /= Bool) => ByteString -> Tensor v t -> m ()
-- | Adds a <a>imageSummary</a> node. The tag argument is intentionally
-- limited to a single value for simplicity.
imageSummary :: (OneOf '[Word8, Word16, Float] t, MonadBuild m) => ByteString -> Tensor v t -> m ()
-- | Adds a <a>scalarSummary</a> node.
scalarSummary :: (TensorType t, t /= ByteString, t /= Bool, MonadBuild m) => ByteString -> Tensor v t -> m ()
-- | Merge all summaries accumulated in the <a>Build</a> into one summary.
mergeAllSummaries :: MonadBuild m => m SummaryTensor