Fork me on GitHub
Arraymancer Technical reference Tutorial Spellbook (How-To's) Under the hood

Module gates_blas

Types

MatMulGate[TT] = ref object of Gate[TT]
  a: Variable[TT]
  b: Variable[TT]
TODO: generalize to C <- alpha AB + C   Source Edit

Procs

proc `*`[TT](a, b: Variable[TT]): Variable[TT]
  Source Edit

Methods

method forward[TT](self: MatMulGate[TT]; a, b: Variable[TT]): Variable[TT] {.
inline, locks: 0
.}
  Source Edit
method backward[TT](self: MatMulGate[TT]; gradient: TT): SmallDiffs[TT] {.
noInit, inline, locks: 0
.}
  Source Edit