Compiler.Xla.HLO.Builder.Lib.PRNG

For internal spidr use only.

Definitions

data BitGenerator : Type

Totality: total
Visibility: public export
Constructors:

ThreeFry : BitGenerator

Philox : BitGenerator

Hint: 

Cast BitGenerator Int

record RngOutput : Type

Totality: total
Visibility: public export
Constructor: 

MkRngOutput : XlaOp -> XlaOp -> RngOutput

Projections:

.state : RngOutput -> XlaOp

.value : RngOutput -> XlaOp

.value : RngOutput -> XlaOp

Totality: total
Visibility: public export

value : RngOutput -> XlaOp

Totality: total
Visibility: public export

.state : RngOutput -> XlaOp

Totality: total
Visibility: public export

state : RngOutput -> XlaOp

Totality: total
Visibility: public export

uniformFloatingPointDistribution : HasIO io => XlaOp -> XlaOp -> BitGenerator -> XlaOp -> XlaOp -> Shape -> io RngOutput

Totality: total
Visibility: export

normalFloatingPointDistribution : HasIO io => XlaOp -> XlaOp -> BitGenerator -> Shape -> io RngOutput

Totality: total
Visibility: export