package owl-base

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

Module Graph.NeuronSource

Sourcetype arr = A.arr
Sourcetype elt = A.elt
Sourcetype t = Owl_algodiff_generic.Make(A).t =
  1. | F of float
  2. | Arr of arr
  3. | DF of t * t * int
  4. | DR of t * t ref * trace_op * int ref * int
Sourceand trace_op = Owl_algodiff_generic.Make(A).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. | Reshape_D of t
  86. | Maxpool1D_D of t * Owl_types.padding * int array * int array
  87. | Maxpool2D_D of t * Owl_types.padding * int array * int array
  88. | Maxpool3D_D of t * Owl_types.padding * int array * int array
  89. | Avgpool1D_D of t * Owl_types.padding * int array * int array
  90. | Avgpool2D_D of t * Owl_types.padding * int array * int array
  91. | Avgpool3D_D of t * Owl_types.padding * int array * int array
  92. | Tr_Conv1D_D_D of t * t * int array
  93. | Tr_Conv1D_D_C of t * t * int array
  94. | Tr_Conv1D_C_D of t * t * int array
  95. | Tr_Conv2D_D_D of t * t * int array
  96. | Tr_Conv2D_D_C of t * t * int array
  97. | Tr_Conv2D_C_D of t * t * int array
  98. | Tr_Conv3D_D_D of t * t * int array
  99. | Tr_Conv3D_D_C of t * t * int array
  100. | Tr_Conv3D_C_D of t * t * int array
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 : ?axis:int -> t -> int -> t
Sourceval repeat_axes : int array -> t -> int array -> t
Sourceval pack_arr : arr -> t
Sourceval unpack_arr : t -> arr
Sourceval pack_flt : float -> t
Sourceval unpack_flt : t -> float
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 Init : sig ... end
Sourcemodule Input : sig ... end
Sourcemodule Activation : sig ... end
Sourcemodule Linear : sig ... end
Sourcemodule LinearNoBias : sig ... end
Sourcemodule Recurrent : sig ... end
Sourcemodule LSTM : sig ... end
Sourcemodule GRU : sig ... end
Sourcemodule Conv1D : sig ... end
Sourcemodule TransposeConv1D : sig ... end
Sourcemodule Conv2D : sig ... end
Sourcemodule TransposeConv2D : sig ... end
Sourcemodule Conv3D : sig ... end
Sourcemodule TransposeConv3D : sig ... end
Sourcemodule FullyConnected : sig ... end
Sourcemodule MaxPool1D : sig ... end
Sourcemodule MaxPool2D : sig ... end
Sourcemodule AvgPool1D : sig ... end
Sourcemodule AvgPool2D : sig ... end
Sourcemodule GlobalMaxPool1D : sig ... end
Sourcemodule GlobalMaxPool2D : sig ... end
Sourcemodule GlobalAvgPool1D : sig ... end
Sourcemodule GlobalAvgPool2D : sig ... end
Sourcemodule UpSampling1D : sig ... end
Sourcemodule UpSampling2D : sig ... end
Sourcemodule UpSampling3D : sig ... end
Sourcemodule Padding1D : sig ... end
Sourcemodule Padding2D : sig ... end
Sourcemodule Padding3D : sig ... end
Sourcemodule Lambda : sig ... end
Sourcemodule Dropout : sig ... end
Sourcemodule Reshape : sig ... end
Sourcemodule Flatten : sig ... end
Sourcemodule Add : sig ... end
Sourcemodule Mul : sig ... end
Sourcemodule Dot : sig ... end
Sourcemodule Max : sig ... end
Sourcemodule Average : sig ... end
Sourcemodule Concatenate : sig ... end
Sourcemodule Normalisation : sig ... end
Sourcemodule GaussianNoise : sig ... end
Sourcemodule GaussianDropout : sig ... end
Sourcemodule AlphaDropout : sig ... end
Sourcemodule Embedding : sig ... end
Sourcemodule Masking : sig ... end
Sourcetype neuron = Owl_neural_neuron.Make(A).neuron =
  1. | Input of Input.neuron_typ
  2. | Linear of Linear.neuron_typ
  3. | LinearNoBias of LinearNoBias.neuron_typ
  4. | Embedding of Embedding.neuron_typ
  5. | LSTM of LSTM.neuron_typ
  6. | GRU of GRU.neuron_typ
  7. | Recurrent of Recurrent.neuron_typ
  8. | Conv1D of Conv1D.neuron_typ
  9. | Conv2D of Conv2D.neuron_typ
  10. | Conv3D of Conv3D.neuron_typ
  11. | TransposeConv1D of TransposeConv1D.neuron_typ
  12. | TransposeConv2D of TransposeConv2D.neuron_typ
  13. | TransposeConv3D of TransposeConv3D.neuron_typ
  14. | FullyConnected of FullyConnected.neuron_typ
  15. | MaxPool1D of MaxPool1D.neuron_typ
  16. | MaxPool2D of MaxPool2D.neuron_typ
  17. | AvgPool1D of AvgPool1D.neuron_typ
  18. | AvgPool2D of AvgPool2D.neuron_typ
  19. | GlobalMaxPool1D of GlobalMaxPool1D.neuron_typ
  20. | GlobalMaxPool2D of GlobalMaxPool2D.neuron_typ
  21. | GlobalAvgPool1D of GlobalAvgPool1D.neuron_typ
  22. | GlobalAvgPool2D of GlobalAvgPool2D.neuron_typ
  23. | Dropout of Dropout.neuron_typ
  24. | Reshape of Reshape.neuron_typ
  25. | Flatten of Flatten.neuron_typ
  26. | Lambda of Lambda.neuron_typ
  27. | Activation of Activation.neuron_typ
  28. | GaussianNoise of GaussianNoise.neuron_typ
  29. | GaussianDropout of GaussianDropout.neuron_typ
  30. | AlphaDropout of AlphaDropout.neuron_typ
  31. | Normalisation of Normalisation.neuron_typ
  32. | Add of Add.neuron_typ
  33. | Mul of Mul.neuron_typ
  34. | Dot of Dot.neuron_typ
  35. | Max of Max.neuron_typ
  36. | Average of Average.neuron_typ
  37. | Concatenate of Concatenate.neuron_typ
Sourceval get_in_out_shape : neuron -> int array * int array
Sourceval get_in_shape : neuron -> int array
Sourceval get_out_shape : neuron -> int array
Sourceval connect : int array array -> neuron -> unit
Sourceval init : neuron -> unit
Sourceval reset : neuron -> unit
Sourceval mktag : int -> neuron -> unit
Sourceval mkpar : neuron -> t array
Sourceval mkpri : neuron -> t array
Sourceval mkadj : neuron -> t array
Sourceval update : neuron -> t array -> unit
Sourceval copy : neuron -> neuron
Sourceval run : t array -> neuron -> t
Sourceval to_string : neuron -> string
Sourceval to_name : neuron -> string