def routed

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

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/matmul.bend as Matrix

364# The operands' dimensions `dims` and `other` decide: two rank-2 operands go
365# to `Matrix.matmul`, every other pair to the plan.
366def routed(+left: Shape.Shape, +right: Shape.Shape, a: Core.Tensor<left>, b: Core.Tensor<right>,
367  +dims: List<&2, U32>, +other: List<&2, U32>) -> Result<&2, &1, Matrix.Error, Core.Dynamic & Core.Dynamic & Core.Dynamic>:
368  dispatched(left, right, a, b, dims, other, matrices(dims, other))