Safe Haskell | None |
---|---|
Language | Haskell2010 |
- withControlDependencies :: Nodes t => t -> Build a -> Build a
- group :: Nodes t => t -> Build ControlNode
- identity :: TensorType a => Tensor v a -> Tensor v a
- noOp :: ControlNode
- named :: TensorType a => Text -> Tensor v a -> Tensor v a
Dependencies
withControlDependencies :: Nodes t => t -> Build a -> Build a Source
Modify a Build
action, such that all new ops rendered in it will depend
on the nodes in the first argument.
group :: Nodes t => t -> Build ControlNode Source
Create an op that groups multiple operations.
When this op finishes, all ops in the input n
have finished. This op has
no output.
Operations
identity :: TensorType a => Tensor v a -> Tensor v a Source
Returns a Tensor
with the same shape and contents as the input.
Does nothing. Only useful as a placeholder for control edges.
named :: TensorType a => Text -> Tensor v a -> Tensor v a Source
Returns a Tensor
with a given name and the same shape and contents as
the input.
TODO(judahjacobson): This breaks when used with uninitialize Tensor Ref
s,
since RefIdentity
doesn't have SetAllowsUninitializedInput(). Look into
whether we can change that op.