def indexing_entries

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

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

1555def indexing_entries(+indexing: Indexing) -> List<&2, U32>:
1556  match indexing:
1557    case IndexSelected{dim, entries}:
1558      entries
1559    case GatheredBy{dim, Gather.Index{shape, entries}}:
1560      entries

Uses (4)

Used by (1)