package eigen

  1. Overview
  2. Docs
val test : (float, 'a, 'b) Bigarray.Genarray.t -> unit
val spatial_conv : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> (float, 'e, 'f) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val spatial_conv_backward_input : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> (float, 'e, 'f) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val spatial_conv_backward_kernel : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> (float, 'e, 'f) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val cuboid_conv : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> (float, 'e, 'f) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val cuboid_conv_backward_input : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> (float, 'e, 'f) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val cuboid_conv_backward_kernel : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> (float, 'e, 'f) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val spatial_max_pooling : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val spatial_avg_pooling : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val cuboid_max_pooling : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val cuboid_avg_pooling : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val spatial_max_pooling_argmax : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> (int64, 'e, 'f) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val spatial_max_pooling_backward : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> (float, 'e, 'f) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val spatial_avg_pooling_backward : (float, 'a, 'b) Bigarray.Genarray.t -> (float, 'c, 'd) Bigarray.Genarray.t -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
OCaml

Innovation. Community. Security.