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

free-tensors!

(free-tensors! node cache)(free-tensors! node)

validate-input-keys

(validate-input-keys nodes input->vals)

with-model-params

(with-model-params model node)