package owl-base

  1. Overview
  2. Docs
Legend:
Library
Module
Module type
Parameter
Class
Class type
module Device : sig ... end
and block = Make_Nested(Owl_computation_engine.Make_Graph(Owl_computation_cpu_device.Make(A))).Graph.Optimiser.Operator.Symbol.Shape.Type.block = {
  1. size : int;
  2. block_id : int;
  3. mutable active : t option;
  4. mutable memory : Device.value;
  5. mutable nodes : t list;
}
and attr = Make_Nested(Owl_computation_engine.Make_Graph(Owl_computation_cpu_device.Make(A))).Graph.Optimiser.Operator.Symbol.Shape.Type.attr = {
  1. mutable op : op;
  2. mutable freeze : bool;
  3. mutable reuse : bool;
  4. mutable state : state;
  5. mutable shape : int array option array;
  6. mutable value : Device.value array;
  7. mutable block : block array option;
}
and op = Make_Nested(Owl_computation_engine.Make_Graph(Owl_computation_cpu_device.Make(A))).Graph.Optimiser.Operator.Symbol.Shape.Type.op =
  1. | Noop
  2. | Var
  3. | Const
  4. | Empty of int array
  5. | Zeros of int array
  6. | Ones of int array
  7. | Create of int array
  8. | Sequential of int array
  9. | Uniform of int array
  10. | Gaussian of int array
  11. | Bernoulli of int array
  12. | Init of int array * int -> elt
  13. | Get of int array
  14. | Set of int array
  15. | GetSlice of int list list
  16. | SetSlice of int list list
  17. | Copy
  18. | Reset
  19. | Reshape of int array
  20. | Reverse
  21. | Tile of int array
  22. | Repeat of int array
  23. | Pad of elt * int list list
  24. | Concatenate of int
  25. | Stack of int
  26. | Split of int * int array
  27. | Draw of int * int
  28. | Map of elt -> elt
  29. | Fold of int * elt -> elt -> elt
  30. | Scan of int * elt -> elt -> elt
  31. | OneHot of int
  32. | OfArray of int array
  33. | Delay of Device.A.arr -> Device.A.arr
  34. | DelayArray of int array * Device.A.arr array -> Device.A.arr
  35. | LazyPrint of int option * int option * bool option * (Device.A.elt -> string) option
  36. | Abs
  37. | Neg
  38. | Floor
  39. | Ceil
  40. | Round
  41. | Sqr
  42. | Sqrt
  43. | Log
  44. | Log2
  45. | Log10
  46. | Exp
  47. | Sin
  48. | Cos
  49. | Tan
  50. | Sinh
  51. | Cosh
  52. | Tanh
  53. | Asin
  54. | Acos
  55. | Atan
  56. | Asinh
  57. | Acosh
  58. | Atanh
  59. | Min of bool * int
  60. | Max of bool * int
  61. | Sum of bool * int
  62. | SumReduce of int array
  63. | Signum
  64. | Sigmoid
  65. | Relu
  66. | Dawsn
  67. | Min'
  68. | Max'
  69. | Sum'
  70. | LogSumExp'
  71. | LogSumExp of bool * int
  72. | L1norm'
  73. | L2norm'
  74. | L2NormSqr'
  75. | ClipByValue
  76. | ClipByL2norm
  77. | Pow
  78. | ScalarPow
  79. | PowScalar
  80. | Atan2
  81. | ScalarAtan2
  82. | Atan2Scalar
  83. | Hypot
  84. | Min2
  85. | Max2
  86. | Add
  87. | Sub
  88. | Mul
  89. | Div
  90. | AddScalar
  91. | SubScalar
  92. | MulScalar
  93. | DivScalar
  94. | ScalarAdd
  95. | ScalarSub
  96. | ScalarMul
  97. | ScalarDiv
  98. | FMA
  99. | EltEqual
  100. | EltNotEqual
  101. | EltLess
  102. | EltGreater
  103. | EltLessEqual
  104. | EltGreaterEqual
  105. | EltEqualScalar
  106. | EltNotEqualScalar
  107. | EltLessScalar
  108. | EltGreaterScalar
  109. | EltLessEqualScalar
  110. | EltGreaterEqualScalar
  111. | Conv1d of Owl_types.padding * int array
  112. | Conv2d of Owl_types.padding * int array
  113. | Conv3d of Owl_types.padding * int array
  114. | TransposeConv1d of Owl_types.padding * int array
  115. | TransposeConv2d of Owl_types.padding * int array
  116. | TransposeConv3d of Owl_types.padding * int array
  117. | DilatedConv1d of Owl_types.padding * int array * int array
  118. | DilatedConv2d of Owl_types.padding * int array * int array
  119. | DilatedConv3d of Owl_types.padding * int array * int array
  120. | MaxPool1d of Owl_types.padding * int array * int array
  121. | MaxPool2d of Owl_types.padding * int array * int array
  122. | MaxPool3d of Owl_types.padding * int array * int array
  123. | AvgPool1d of Owl_types.padding * int array * int array
  124. | AvgPool2d of Owl_types.padding * int array * int array
  125. | AvgPool3d of Owl_types.padding * int array * int array
  126. | UpSampling2d of int array
  127. | Conv1dBackwardInput of int array
  128. | Conv1dBackwardKernel of int array
  129. | Conv2dBackwardInput of int array
  130. | Conv2dBackwardKernel of int array
  131. | Conv3dBackwardInput of int array
  132. | Conv3dBackwardKernel of int array
  133. | TransposeConv1dBackwardInput of int array
  134. | TransposeConv1dBackwardKernel of int array
  135. | TransposeConv2dBackwardInput of int array
  136. | TransposeConv2dBackwardKernel of int array
  137. | TransposeConv3dBackwardInput of int array
  138. | TransposeConv3dBackwardKernel of int array
  139. | DilatedConv1dBackwardInput of int array * int array
  140. | DilatedConv1dBackwardKernel of int array * int array
  141. | DilatedConv2dBackwardInput of int array * int array
  142. | DilatedConv2dBackwardKernel of int array * int array
  143. | DilatedConv3dBackwardInput of int array * int array
  144. | DilatedConv3dBackwardKernel of int array * int array
  145. | MaxPool1dBackward of Owl_types.padding * int array * int array
  146. | MaxPool2dBackward of Owl_types.padding * int array * int array
  147. | MaxPool3dBackward of Owl_types.padding * int array * int array
  148. | AvgPool1dBackward of Owl_types.padding * int array * int array
  149. | AvgPool2dBackward of Owl_types.padding * int array * int array
  150. | AvgPool3dBackward of Owl_types.padding * int array * int array
  151. | UpSampling2dBackward of int array
  152. | RowNum
  153. | ColNum
  154. | Row
  155. | Rows of int array
  156. | CopyRowTo
  157. | CopyColTo
  158. | Dot of bool * bool * elt * elt
  159. | Inv
  160. | Trace
  161. | Transpose of int array
  162. | ToRows
  163. | OfRows
  164. | Scalar_Add
  165. | Scalar_Sub
  166. | Scalar_Mul
  167. | Scalar_Div
  168. | Scalar_Pow
  169. | Scalar_Atan2
  170. | Scalar_Abs
  171. | Scalar_Neg
  172. | Scalar_Sqr
  173. | Scalar_Sqrt
  174. | Scalar_Exp
  175. | Scalar_Log
  176. | Scalar_Log2
  177. | Scalar_Log10
  178. | Scalar_Signum
  179. | Scalar_Floor
  180. | Scalar_Ceil
  181. | Scalar_Round
  182. | Scalar_Sin
  183. | Scalar_Cos
  184. | Scalar_Tan
  185. | Scalar_Sinh
  186. | Scalar_Cosh
  187. | Scalar_Tanh
  188. | Scalar_Asin
  189. | Scalar_Acos
  190. | Scalar_Atan
  191. | Scalar_Asinh
  192. | Scalar_Acosh
  193. | Scalar_Atanh
  194. | Scalar_Relu
  195. | Scalar_Dawsn
  196. | Scalar_Sigmoid
  197. | Fused_Adagrad of float * float
OCaml

Innovation. Community. Security.