flare.compute
backward-pass!
(backward-pass! factory target cache)
(backward-pass! factory target)
(backward-pass! target)
backward-pass through all the parameter nodes associated with the graph computation, will write to :grad
key for all nodes that have gradients (basically non-inputs) in graph
BatchTensorOp
protocol
members
batch-backward-node-pass!
(batch-backward-node-pass! this sig nodes)
compute the batch version of the backward pass
batch-forward-node-pass!
(batch-forward-node-pass! this sig nodes)
compute the batch version of the forward pass
batch-signature
(batch-signature this node)
signature that can be used to group operations
forward-pass!
(forward-pass! factory target)
(forward-pass! factory target cache)
(forward-pass! node)
NOTE: Don’t use this forward-pass will topographic walk through graph writing to :value
key on all compiled nodes. You can then look up and retrieve the tensors associated with any node