def cell_at

bendtorch-core@0.1.0.0 · src/tensor/gather.bend:313

import bendtorch-core@0.1.0.0/src/tensor/gather.bend as Gather

import Base

313# The gradient cells (o, k, r), (o, k + 1, r), ... for `left` more k, added
314# to acc in increasing k.
315# Gradient cell (o, k, r) and the index slot it reads.
316def cell_at(+plan: Plan, +o: Nat, +k: Nat, +r: Nat) -> Nat:
317  Plan{+kind, +outer, +extent, +inner, +length} = plan
318  position(length, inner, o, k, r)

Uses (3)

Used by (3)