TensorBackward.operand_ids: def of bendtorch-core@0.1.0.0
Import: import bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend as TensorBackward
Lines 1072-1106 of bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend: https://jend.leeeet.dev/read.txt?path=bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend&start_line=1072&end_line=1106

import Base

# The operand ids of a node, in operand order.
def operand_ids(node: Node) -> List<&2, Nat>:
  match node:
    case Leaf{}:
      []
    case Add{a, b, left, right}:
      [a, b]
    case Sub{a, b, left, right}:
      [a, b]
    case Mul{a, b, left, right}:
      [a, b]
    case Sum{a, input}:
      [a]
    case SumDims{a, input, kept}:
      [a]
    case Matmul{a, b, left, right}:
      [a, b]
    case MatmulNT{a, b, left, right}:
      [a, b]
    case Relu{a, input}:
      [a]
    case MseLoss{a, b, left, right}:
      [a, b]
    case CrossEntropy{a, input}:
      [a]
    case BatchedMatmul{a, b, left, right}:
      [a, b]
    case Slice{a, input, kept, plan}:
      [a]
    case Gathered{a, input, plan, entries}:
      [a]
    case Constant{}:
      []
    case Frozen{id, inner}:
      operand_ids(inner)

Uses (3):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=List
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend&name=Node