Compiler.Xla.HLO.Builder.XlaBuilder

For internal spidr use only.

Definitions

data XlaBuilder : Type

Totality: total
Visibility: public export
Constructor: 

MkXlaBuilder : GCAnyPtr -> XlaBuilder

data XlaOp : Type

Totality: total
Visibility: public export
Constructor: 

MkXlaOp : GCAnyPtr -> XlaOp

delete : HasIO io => AnyPtr -> io ()

Totality: total
Visibility: export

delete : HasIO io => AnyPtr -> io ()

Totality: total
Visibility: export

mkXlaBuilder : HasIO io => String -> io XlaBuilder

Totality: total
Visibility: export

name : HasIO io => XlaBuilder -> io String

Totality: total
Visibility: export

createSubBuilder : HasIO io => XlaBuilder -> String -> io XlaBuilder

Totality: total
Visibility: export

build : HasIO io => XlaBuilder -> XlaOp -> io XlaComputation

Totality: total
Visibility: export

opToString : XlaBuilder -> XlaOp -> String

Totality: total
Visibility: export

mkXlaOpArray : HasIO io => List XlaOp -> io XlaOpArray

Totality: total
Visibility: export

parameter : HasIO io => XlaBuilder -> Nat -> Shape -> String -> io XlaOp

Totality: total
Visibility: export

constantLiteral : HasIO io => XlaBuilder -> Literal -> io XlaOp

Totality: total
Visibility: export

broadcast : HasIO io => XlaOp -> List Nat -> io XlaOp

Totality: total
Visibility: export

broadcastInDim : HasIO io => XlaOp -> List Nat -> List Nat -> io XlaOp

Totality: total
Visibility: export

reshape : HasIO io => XlaOp -> List Nat -> List Nat -> io XlaOp

Totality: total
Visibility: export

slice : HasIO io => XlaOp -> List Nat -> List Nat -> List Nat -> io XlaOp

Totality: total
Visibility: export

dynamicSlice : HasIO io => XlaOp -> List XlaOp -> List Nat -> io XlaOp

Totality: total
Visibility: export

concatInDim : HasIO io => XlaBuilder -> (operands : List XlaOp) -> {auto 0 _ : NonEmpty operands} -> Nat -> io XlaOp

Totality: total
Visibility: export

select : HasIO io => XlaOp -> XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

tuple : HasIO io => XlaBuilder -> List XlaOp -> io XlaOp

Totality: total
Visibility: export

getTupleElement : HasIO io => XlaOp -> Nat -> io XlaOp

Totality: total
Visibility: export

eq : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

ne : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

ge : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

gt : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

lt : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

le : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

dot : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

dotGeneral : HasIO io => XlaOp -> XlaOp -> DotDimensionNumbers -> io XlaOp

Totality: total
Visibility: export

data Transpose : Type

Totality: total
Visibility: public export
Constructors:

NoTranspose : Transpose

Transpose_ : Transpose

Adjoint : Transpose

triangularSolve : HasIO io => XlaOp -> XlaOp -> Bool -> Bool -> Bool -> Transpose -> io XlaOp

Totality: total
Visibility: export

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

Totality: total
Visibility: export

add : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

sub : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

mul : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

div : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export
Fixity Declaration: infixl operator, level 9

rem : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

max : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

min : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

and : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

or : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

unaryOp : HasIO io => (GCAnyPtr -> PrimIO AnyPtr) -> XlaOp -> io XlaOp

Totality: total
Visibility: export

not : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

reduce : HasIO io => XlaOp -> XlaOp -> XlaComputation -> List Nat -> io XlaOp

Totality: total
Visibility: export

abs : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

exp : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

floor : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

ceil : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

log : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

logistic : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

cos : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

sin : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

tanh : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

sqrt : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

pow : HasIO io => XlaOp -> XlaOp -> io XlaOp

Totality: total
Visibility: export

iota : HasIO io => XlaBuilder -> Shape -> Nat -> io XlaOp

Totality: total
Visibility: export

convertElementType : (HasIO io, Primitive dtype) => XlaOp -> io XlaOp

Totality: total
Visibility: export

neg : HasIO io => XlaOp -> io XlaOp

Totality: total
Visibility: export

transpose : HasIO io => XlaOp -> List Nat -> io XlaOp

Totality: total
Visibility: export

rev : HasIO io => XlaOp -> List Nat -> io XlaOp

Totality: total
Visibility: export

sort : HasIO io => List XlaOp -> XlaComputation -> Nat -> Bool -> io XlaOp

Totality: total
Visibility: export

map : HasIO io => XlaBuilder -> List XlaOp -> XlaComputation -> List Nat -> io XlaOp

Totality: total
Visibility: export

data RandomAlgorithm : Type

Totality: total
Visibility: public export
Constructors:

RngDefault : RandomAlgorithm

RngThreeFry : RandomAlgorithm

RngPhilox : RandomAlgorithm

rngBitGenerator : HasIO io => RandomAlgorithm -> XlaOp -> Shape -> io XlaOp

Totality: total
Visibility: export

conditional : HasIO io => XlaOp -> XlaOp -> XlaComputation -> XlaOp -> XlaComputation -> io XlaOp

Totality: total
Visibility: export