Compiler.Xla.HLO.Builder.Lib.Matrix

For internal spidr use only.

Definitions

identityMatrix : HasIO io => Primitive dtype => XlaBuilder -> Nat -> Nat -> io XlaOp

Totality: total
Visibility: export

getMatrixDiagonal : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

triangle : HasIO io => XlaOp -> Bool -> io XlaOp

Totality: total
Visibility: export