def multiply_source

bendtorch-core@0.1.0.0 · src/tensor/matmul_PROOF.bend:1610

import bendtorch-core@0.1.0.0/src/tensor/matmul_PROOF.bend as MatmulPROOF
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
import bendtorch-core@0.1.0.0/src/tensor/matmul_LAWS.bend as Laws

1610def multiply_source(+left: Shape.Shape, +right: Shape.Shape, +plan: Matrix.Plan<left, right>, a: Core.Tensor<left>, b: Core.Tensor<right>) ->
1611  Laws.SourceMatmul<Core.Packed{left, a}, Core.Packed{right, b}, Matrix.parallel_multiply(left, right, plan, a, b)>:
1612  match plan:
1613    case Matrix.Plan{+output, +rows, +inner, +right_inner, +columns, le, re, oe, compatible}:
1614      allocated_source(left, right, output, rows, inner, right_inner, columns, le, re, oe, compatible, a, b, Core.new(output, 0.0))

Uses (8)

Used by (1)