mirror of
https://github.com/tensorflow/haskell.git
synced 2024-11-17 00:19:44 +01:00
26 lines
875 B
Text
26 lines
875 B
Text
-- Hoogle documentation, generated by Haddock
|
|
-- See Hoogle, http://www.haskell.org/hoogle/
|
|
|
|
|
|
-- | Basic access to TensorFlow queues.
|
|
--
|
|
-- Please see README.md
|
|
@package tensorflow-queue
|
|
@version 0.1.0.0
|
|
|
|
|
|
-- | Queues in TensorFlow graph. Very limited support for now.
|
|
module TensorFlow.Queue
|
|
|
|
-- | A queue carrying tuples. The underlying structure is more versatile
|
|
-- and can be made to support arbitrary tuples.
|
|
data Queue2 a b
|
|
|
|
-- | Creates a new queue with the given capacity and shared name.
|
|
makeQueue2 :: (TensorType a, TensorType b) => Int64 -> ByteString -> Build (Queue2 a b)
|
|
|
|
-- | Adds the given values to the queue.
|
|
enqueue :: (TensorType a, TensorType b) => Queue2 a b -> Tensor v1 a -> Tensor v2 b -> Build ControlNode
|
|
|
|
-- | Retrieves the values from the queue.
|
|
dequeue :: (TensorType a, TensorType b) => Queue2 a b -> Build (Tensor Ref a, Tensor Ref b)
|