package owl-base

  1. Overview
  2. Docs
Legend:
Library
Module
Module type
Parameter
Class
Class type

Parameters

Signature

module A = A
type 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
and 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. | Add_Row_D_D of t * t * int
  69. | Add_Row_D_C of t * t * int
  70. | Add_Row_C_D of t * t * int
  71. | Get_Row_D of t * int
  72. | Of_Rows_D of t array
  73. | Concat_D_D of t * t * int
  74. | Concat_D_C of t * t * int
  75. | Concat_C_D of t * t * int
  76. | Conv1D_D_D of t * t * int array
  77. | Conv1D_D_C of t * t * int array
  78. | Conv1D_C_D of t * t * int array
  79. | Conv2D_D_D of t * t * int array
  80. | Conv2D_D_C of t * t * int array
  81. | Conv2D_C_D of t * t * int array
  82. | Conv3D_D_D of t * t * int array
  83. | Conv3D_D_C of t * t * int array
  84. | Conv3D_C_D of t * t * int array
  85. | Di_Conv1D_D_D of t * t * int array * int array
  86. | Di_Conv1D_D_C of t * t * int array * int array
  87. | Di_Conv1D_C_D of t * t * int array * int array
  88. | Di_Conv2D_D_D of t * t * int array * int array
  89. | Di_Conv2D_D_C of t * t * int array * int array
  90. | Di_Conv2D_C_D of t * t * int array * int array
  91. | Di_Conv3D_D_D of t * t * int array * int array
  92. | Di_Conv3D_D_C of t * t * int array * int array
  93. | Di_Conv3D_C_D of t * t * int array * int array
  94. | Tr_Conv1D_D_D of t * t * int array
  95. | Tr_Conv1D_D_C of t * t * int array
  96. | Tr_Conv1D_C_D of t * t * int array
  97. | Tr_Conv2D_D_D of t * t * int array
  98. | Tr_Conv2D_D_C of t * t * int array
  99. | Tr_Conv2D_C_D of t * t * int array
  100. | Tr_Conv3D_D_D of t * t * int array
  101. | Tr_Conv3D_D_C of t * t * int array
  102. | Tr_Conv3D_C_D of t * t * int array
  103. | Reshape_D of t
  104. | Maxpool1D_D of t * Owl_types.padding * int array * int array
  105. | Maxpool2D_D of t * Owl_types.padding * int array * int array
  106. | Maxpool3D_D of t * Owl_types.padding * int array * int array
  107. | Avgpool1D_D of t * Owl_types.padding * int array * int array
  108. | Avgpool2D_D of t * Owl_types.padding * int array * int array
  109. | Avgpool3D_D of t * Owl_types.padding * int array * int array
val _global_tag : int ref
val tag : unit -> int
val cmp_tag : 'a -> 'a -> int
val reset_zero : t -> t
val primal : t -> t
val primal' : t -> t
val zero : t -> t
val tangent : t -> t
val adjref : t -> t ref
val adjval : t -> t
val shape : t -> int array
val row_num : t -> int
val col_num : t -> int
val numel : t -> int
val clip_by_value : amin:A.elt -> amax:A.elt -> t -> t
val clip_by_l2norm : A.elt -> t -> t
val copy_primal' : t -> t
val tile : t -> int array -> t
val repeat : t -> int array -> t
val pack_elt : A.elt -> t
val unpack_elt : t -> A.elt
val pack_flt : float -> t
val _f : float -> t
val unpack_flt : t -> float
val pack_arr : A.arr -> t
val unpack_arr : t -> A.arr
val deep_info : t -> string
val type_info : t -> string
val error_binop : string -> t -> t -> 'a
val error_uniop : string -> t -> 'a
module Maths : sig ... end
val reverse_reset : t -> unit
val _shrink : t -> t -> t
val reverse_push : t -> t -> unit
val reverse_prop : t -> t -> unit
val make_forward : t -> t -> int -> t
val make_reverse : t -> int -> t
val diff' : (t -> t) -> t -> t * t
val diff : (t -> t) -> t -> t
val grad' : (t -> t) -> t -> t * t
val grad : (t -> t) -> t -> t
val jacobianv' : (t -> t) -> t -> t -> t * t
val jacobianv : (t -> t) -> t -> t -> t
val jacobianTv' : (t -> t) -> t -> t -> t * t
val jacobianTv : (t -> t) -> t -> t -> t
val jacobian' : (t -> t) -> t -> t * t
val jacobian : (t -> t) -> t -> t
val gradhessian : (t -> t) -> t -> t * t
val gradhessian' : (t -> t) -> t -> t * t * t
val hessian : (t -> t) -> t -> t
val hessian' : (t -> t) -> t -> t * t
val gradhessianv' : (t -> t) -> t -> t -> t * t * t
val gradhessianv : (t -> t) -> t -> t -> t * t
val hessianv' : (t -> t) -> t -> t -> t * t
val hessianv : (t -> t) -> t -> t -> t
val laplacian : (t -> t) -> t -> t
val laplacian' : (t -> t) -> t -> t * t
module Mat : sig ... end
module Arr : sig ... end
val _traverse_trace : t list -> (t, int * string * t list) Hashtbl.t
val _convert_terminal_output : (t, int * string * t list) Hashtbl.t -> string
val _convert_dot_output : (t, int * string * t list) Hashtbl.t -> string
val to_trace : t list -> string
val to_dot : t list -> string
val pp_num : Format.formatter -> t -> unit
OCaml

Innovation. Community. Security.