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