~/jendhub

def plan_dims

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul.bend:184

import bendtorch-core@0.1.0.0/src/tensor/batched_matmul.bend as BatchedMatmul
import Base
import bendtorch-core@0.1.0.0/src/tensor/matmul.bend as Matrix

184# The plan of left @ right for operands of dimensions `left` and `right`,
185# each of rank 1 to 4: the left operand's rank is checked first, then the
186# right one's, the inner extents, the batch dimensions and the output's
187# capacity.
188def plan_dims(left: List<&2, U32>, right: List<&2, U32>) -> Result<&2, &2, Matrix.Error, Plan>:
189  left_parsed(right, left_operand(left))

Uses (7)

Used by (3)