def freeze

bendtorch-core@0.1.0.0 · src/autograd/tensor_backward.bend:1115

import bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend as TensorBackward
import Base

1115# node wrapped in Frozen{id, .} for each operand id that is a constant of the
1116# newest-first list of count nodes, the first operand outermost.
1117def freeze(ids: List<&2, Nat>, +nodes: List<&2, Node>, +count: Nat, node: Node) -> Node:
1118  match ids:
1119    case Nil{}:
1120      node
1121    case Con{+id, rest}:
1122      frozen_if(id, freeze(rest, nodes, count, node), constant_at(nodes, count, id, at_head(count, id)))

Uses (6)

Used by (1)