type Walk

bendtorch-core@0.1.0.0 · src/tensor/elementwise.bend:362

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

362# What every cell of a build needs: the operation, the broadcast index map
363# and the slot each input reads (`left_slot`, `right_slot`), so that a build
364# whose inputs have the output's shape or rank 0 skips the walk. Each thread
365# below the forks computes its own once.
366type Walk is Data:
367  Walk{operation: Operation, -le: List<&2, Nat>, -re: List<&2, Nat>, -oe: List<&2, Nat>, map: Shape.Index<le, re, oe>,
368    left: Slot, right: Slot}