Idris2Doc : Compiler.IR

Compiler.IR

(source)
For internal spidr use only.

Reexports

import public Compiler.Stablehlo.Dialect.StablehloEnums

Definitions

data ValueType : Type
Totality: total
Visibility: public export
Constructor: 
TensorType : Shape -> (0 dtype : Type) -> Primitive dtype => ValueType

Hint: 
Show ValueType
data Value : Type
Totality: total
Visibility: public export
Constructor: 
V : Nat -> Op -> Value

Hint: 
Show Value
data Env : Type
Totality: total
Visibility: export
Constructor: 
MkEnv : Nat -> List (Nat, Op) -> Env
empty : Env
Totality: total
Visibility: export
emptyFrom : Env -> Env
Totality: total
Visibility: export
updateCounterFrom : Env -> State Env ()
Totality: total
Visibility: export
toList : Env -> List (Nat, Op)
Totality: total
Visibility: export
counter : Env -> Nat
Totality: total
Visibility: export
record Fn : Nat -> Type
Totality: total
Visibility: public export
Constructor: 
MkFn : Nat -> Vect arity ValueType -> Vect resultCount ValueType -> Vect resultCount Value -> Env -> Fn arity

Projections:
.env : Fn arity -> Env
.paramTypes : Fn arity -> Vect arity ValueType
0 .resultCount : Fn arity -> Nat
.resultTypes : ({rec:0} : Fn arity) -> Vect (resultCount {rec:0}) ValueType
.results : ({rec:0} : Fn arity) -> Vect (resultCount {rec:0}) Value
.tag : Fn arity -> Nat

Hint: 
Show (Fn arity)
.tag : Fn arity -> Nat
Totality: total
Visibility: public export
tag : Fn arity -> Nat
Totality: total
Visibility: public export
.paramTypes : Fn arity -> Vect arity ValueType
Totality: total
Visibility: public export
paramTypes : Fn arity -> Vect arity ValueType
Totality: total
Visibility: public export
.resultTypes : ({rec:0} : Fn arity) -> Vect (resultCount {rec:0}) ValueType
Totality: total
Visibility: public export
resultTypes : ({rec:0} : Fn arity) -> Vect (resultCount {rec:0}) ValueType
Totality: total
Visibility: public export
.results : ({rec:0} : Fn arity) -> Vect (resultCount {rec:0}) Value
Totality: total
Visibility: public export
results : ({rec:0} : Fn arity) -> Vect (resultCount {rec:0}) Value
Totality: total
Visibility: public export
.env : Fn arity -> Env
Totality: total
Visibility: public export
env : Fn arity -> Env
Totality: total
Visibility: public export
data BinaryOp : Type
Totality: total
Visibility: public export
Constructors:
Compare : ComparisonDirection -> BinaryOp
And : BinaryOp
Or : BinaryOp
Add : BinaryOp
Sub : BinaryOp
Mul : BinaryOp
Div : BinaryOp
Rem : BinaryOp
Pow : BinaryOp
Min : BinaryOp
Max : BinaryOp
ShiftRightLogical : BinaryOp

Hint: 
Show BinaryOp
data UnaryOp : Type
Totality: total
Visibility: public export
Constructors:
Not : UnaryOp
Neg : UnaryOp
Ceil : UnaryOp
Floor : UnaryOp
Abs : UnaryOp
Log : UnaryOp
Exp : UnaryOp
Logistic : UnaryOp
Sqrt : UnaryOp
Sin : UnaryOp
Cos : UnaryOp
Tan : UnaryOp
Tanh : UnaryOp
Erf : UnaryOp
ErfInv : UnaryOp
Square : UnaryOp
Asin : UnaryOp
Acos : UnaryOp
Atan : UnaryOp
Sinh : UnaryOp
Cosh : UnaryOp
Asinh : UnaryOp
Acosh : UnaryOp
Atanh : UnaryOp

Hint: 
Show UnaryOp
data Op : Type
Totality: total
Visibility: public export
Constructors:
BoundSet : Nat -> Op
Lit : PrimitiveRW dtype ty => Literal shape ty -> Op
Grad : Shape -> Fn 1 -> Value -> Op
MinValue : (0 dtype : Type) -> Primitive dtype => Op
MaxValue : (0 dtype : Type) -> Primitive dtype => Op
MinFiniteFloat : Op
MaxFiniteFloat : Op
Iota : Primitive dtype => Shape -> Nat -> Op
BitCastConvert : Primitive dtype => Shape -> Value -> Op
Convert : Primitive dtype => Shape -> Value -> Op
Reshape : Primitive dtype => Shape -> Value -> Op
Slice : List Nat -> List Nat -> List Nat -> Value -> Op
DynamicSlice : List Value -> List Nat -> Value -> Op
Concat : Nat -> Vect (S n) Value -> Op
Transpose : List Nat -> Value -> Op
Broadcast : Primitive dtype => Shape -> Shape -> Value -> Op
Map : Fn arity -> Vect arity Value -> ValueType -> Shape -> Op
Reduce : Fn (n + n) -> Vect n Value -> List Nat -> Vect n Value -> Op
Sort : Fn 2 -> Nat -> Bool -> Value -> Op
Reverse : List Nat -> Value -> Op
BinaryElementwise : BinaryOp -> Value -> Value -> Op
UnaryElementwise : UnaryOp -> Value -> Op
Select : Value -> Value -> Value -> Op
While : Fn n -> Fn n -> Vect n Value -> Op
If : ValueType -> Value -> Fn 0 -> Fn 0 -> Op
DotGeneral : List Nat -> List Nat -> List Nat -> List Nat -> ValueType -> Value -> Value -> Op
Cholesky : Value -> Op
TriangularSolve : Value -> Value -> Bool -> Op
Rng : Value -> ValueType -> Op

Hints:
Show Op
Taggable Op
tagOp : Monad m => Op -> StateT Env m Op
Totality: total
Visibility: export
reserve : State Env Nat
Totality: total
Visibility: export