package owl-base

  1. Overview
  2. Docs
package owl-base
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module Owl_algodiff_generic.MakeSource

Parameters

Signature

module A = A
Sourcetype t =
  1. | F of A.elt
  2. | Arr of A.arr
  3. | DF of t * t * int
  4. | DR of t * t ref * trace_op * int ref * int * int ref
Sourceand trace_op =
  1. | Noop
  2. | Add_D_D of t * t
  3. | Add_D_C of t * t
  4. | Add_C_D of t * t
  5. | Sub_D_D of t * t
  6. | Sub_D_C of t * t
  7. | Sub_C_D of t * t
  8. | Mul_D_D of t * t
  9. | Mul_D_C of t * t
  10. | Mul_C_D of t * t
  11. | Div_D_D of t * t
  12. | Div_D_C of t * t
  13. | Div_C_D of t * t
  14. | Pow_D_D of t * t
  15. | Pow_D_C of t * t
  16. | Pow_C_D of t * t
  17. | Atan2_D_D of t * t
  18. | Atan2_D_C of t * t
  19. | Atan2_C_D of t * t
  20. | Neg_D of t
  21. | Abs_D of t
  22. | Signum_D of t
  23. | Floor_D of t
  24. | Ceil_D of t
  25. | Round_D of t
  26. | Sqr_D of t
  27. | Sqrt_D of t
  28. | Log_D of t
  29. | Log2_D of t
  30. | Log10_D of t
  31. | Exp_D of t
  32. | Sin_D of t
  33. | Cos_D of t
  34. | Tan_D of t
  35. | Sinh_D of t
  36. | Cosh_D of t
  37. | Tanh_D of t
  38. | Asin_D of t
  39. | Acos_D of t
  40. | Atan_D of t
  41. | Asinh_D of t
  42. | Acosh_D of t
  43. | Atanh_D of t
  44. | Get_Item of t * int * int
  45. | SetI_D_D of t * int * int * t
  46. | SetI_D_C of t * int * int * t
  47. | SetI_C_D of t * int * int * t
  48. | AddI_D_D of t * int * int * t
  49. | AddI_D_C of t * int * int * t
  50. | AddI_C_D of t * int * int * t
  51. | Get_Slice_D of t * int list list
  52. | Set_Slice_D_D of t * t * int list list
  53. | Set_Slice_D_C of t * t * int list list
  54. | Set_Slice_C_D of t * t * int list list
  55. | Sum_D of t
  56. | Sum__D of t * int
  57. | Sum___D of t * int array
  58. | Dot_D_D of t * t
  59. | Dot_D_C of t * t
  60. | Dot_C_D of t * t
  61. | Trans_D of t
  62. | L1Norm_D of t
  63. | L2Norm_D of t
  64. | L2NormS_D of t
  65. | Sigmoid_D of t
  66. | Relu_D of t
  67. | Inv_D of t
  68. | Logdet_D of t
  69. | Chol_D of t * bool
  70. | QR_D of t * (t ref * t ref) * (t ref * t ref)
  71. | Svd_D of t * (t ref * t ref * t ref) * (t ref * t ref * t ref) * bool
  72. | Lyapunov_D_D of t * t
  73. | Lyapunov_C_D of t * t
  74. | Lyapunov_D_C of t * t
  75. | Discrete_Lyapunov_D_D of t * t
  76. | Discrete_Lyapunov_C_D of t * t
  77. | Discrete_Lyapunov_D_C of t * t
  78. | Diag_D of int * t
  79. | Diagm_D of int * t
  80. | Tril_D of int * t
  81. | Triu_D of int * t
  82. | Trace_D of t
  83. | Add_Row_D_D of t * t * int
  84. | Add_Row_D_C of t * t * int
  85. | Add_Row_C_D of t * t * int
  86. | Get_Row_D of t * int
  87. | Of_Rows_D of t array
  88. | Of_Arrays_D of t array array * (int * int) list
  89. | Concat_D_D of t * t * int
  90. | Concat_D_C of t * t * int
  91. | Concat_C_D of t * t * int
  92. | Split_D of t * int * t ref array
  93. | Concatenate_D of t array * int * int list
  94. | Conv1D_D_D of t * t * int array
  95. | Conv1D_D_C of t * t * int array
  96. | Conv1D_C_D of t * t * int array
  97. | Conv2D_D_D of t * t * int array
  98. | Conv2D_D_C of t * t * int array
  99. | Conv2D_C_D of t * t * int array
  100. | Conv3D_D_D of t * t * int array
  101. | Conv3D_D_C of t * t * int array
  102. | Conv3D_C_D of t * t * int array
  103. | Di_Conv1D_D_D of t * t * int array * int array
  104. | Di_Conv1D_D_C of t * t * int array * int array
  105. | Di_Conv1D_C_D of t * t * int array * int array
  106. | Di_Conv2D_D_D of t * t * int array * int array
  107. | Di_Conv2D_D_C of t * t * int array * int array
  108. | Di_Conv2D_C_D of t * t * int array * int array
  109. | Di_Conv3D_D_D of t * t * int array * int array
  110. | Di_Conv3D_D_C of t * t * int array * int array
  111. | Di_Conv3D_C_D of t * t * int array * int array
  112. | Tr_Conv1D_D_D of t * t * int array
  113. | Tr_Conv1D_D_C of t * t * int array
  114. | Tr_Conv1D_C_D of t * t * int array
  115. | Tr_Conv2D_D_D of t * t * int array
  116. | Tr_Conv2D_D_C of t * t * int array
  117. | Tr_Conv2D_C_D of t * t * int array
  118. | Tr_Conv3D_D_D of t * t * int array
  119. | Tr_Conv3D_D_C of t * t * int array
  120. | Tr_Conv3D_C_D of t * t * int array
  121. | Reshape_D of t
  122. | Maxpool1D_D of t * Owl_types.padding * int array * int array
  123. | Maxpool2D_D of t * Owl_types.padding * int array * int array
  124. | Maxpool3D_D of t * Owl_types.padding * int array * int array
  125. | Avgpool1D_D of t * Owl_types.padding * int array * int array
  126. | Avgpool2D_D of t * Owl_types.padding * int array * int array
  127. | Avgpool3D_D of t * Owl_types.padding * int array * int array
  128. | UpSampling2D_D of t * int array
  129. | PAD_D of t * int list list
Sourceval _global_tag : int ref
Sourceval tag : unit -> int
Sourceval cmp_tag : 'a -> 'a -> int
Sourceval reset_zero : t -> t
Sourceval primal : t -> t
Sourceval primal' : t -> t
Sourceval zero : t -> t
Sourceval tangent : t -> t
Sourceval adjref : t -> t ref
Sourceval adjval : t -> t
Sourceval shape : t -> int array
Sourceval row_num : t -> int
Sourceval col_num : t -> int
Sourceval numel : t -> int
Sourceval clip_by_value : amin:A.elt -> amax:A.elt -> t -> t
Sourceval clip_by_l2norm : A.elt -> t -> t
Sourceval copy_primal' : t -> t
Sourceval tile : t -> int array -> t
Sourceval repeat : t -> int array -> t
Sourceval pack_elt : A.elt -> t
Sourceval unpack_elt : t -> A.elt
Sourceval pack_flt : float -> t
Sourceval _f : float -> t
Sourceval unpack_flt : t -> float
Sourceval pack_arr : A.arr -> t
Sourceval unpack_arr : t -> A.arr
Sourceval deep_info : t -> string
Sourceval type_info : t -> string
Sourceval error_binop : string -> t -> t -> 'a
Sourceval error_uniop : string -> t -> 'a
Sourcemodule Maths : sig ... end
Sourceval reverse_reset : t -> unit
Sourceval _shrink : t -> t -> t
Sourceval reverse_push : t -> t -> unit
Sourceval reverse_prop : t -> t -> unit
Sourceval make_forward : t -> t -> int -> t
Sourceval make_reverse : t -> int -> t
Sourceval diff' : (t -> t) -> t -> t * t
Sourceval diff : (t -> t) -> t -> t
Sourceval grad' : (t -> t) -> t -> t * t
Sourceval grad : (t -> t) -> t -> t
Sourceval jacobianv' : (t -> t) -> t -> t -> t * t
Sourceval jacobianv : (t -> t) -> t -> t -> t
Sourceval jacobianTv' : (t -> t) -> t -> t -> t * t
Sourceval jacobianTv : (t -> t) -> t -> t -> t
Sourceval jacobian' : (t -> t) -> t -> t * t
Sourceval jacobian : (t -> t) -> t -> t
Sourceval gradhessian : (t -> t) -> t -> t * t
Sourceval gradhessian' : (t -> t) -> t -> t * t * t
Sourceval hessian : (t -> t) -> t -> t
Sourceval hessian' : (t -> t) -> t -> t * t
Sourceval gradhessianv' : (t -> t) -> t -> t -> t * t * t
Sourceval gradhessianv : (t -> t) -> t -> t -> t * t
Sourceval hessianv' : (t -> t) -> t -> t -> t * t
Sourceval hessianv : (t -> t) -> t -> t -> t
Sourceval laplacian : (t -> t) -> t -> t
Sourceval laplacian' : (t -> t) -> t -> t * t
Sourcemodule Mat : sig ... end
Sourcemodule Arr : sig ... end
Sourceval _traverse_trace : t list -> (t, int * string * t list) Hashtbl.t
Sourceval _convert_terminal_output : (t, int * string * t list) Hashtbl.t -> string
Sourceval _convert_dot_output : (t, int * string * t list) Hashtbl.t -> string
Sourceval to_trace : t list -> string
Sourceval to_dot : t list -> string
Sourceval pp_num : Format.formatter -> t -> unit
Sourcemodule FDGrad_test : sig ... end
OCaml

Innovation. Community. Security.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events OCaml Planet Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct