Module type Owl_dense_ndarray_intf.NN

include Owl_base_dense_ndarray_intf.NN
type arr
val conv1d : ?padding:Owl_types_common.padding -> arr -> arr -> int array -> arr
val conv2d : ?padding:Owl_types_common.padding -> arr -> arr -> int array -> arr
val conv3d : ?padding:Owl_types_common.padding -> arr -> arr -> int array -> arr
val dilated_conv1d : ?padding:Owl_types_common.padding -> arr -> arr -> int array -> int array -> arr
val dilated_conv2d : ?padding:Owl_types_common.padding -> arr -> arr -> int array -> int array -> arr
val dilated_conv3d : ?padding:Owl_types_common.padding -> arr -> arr -> int array -> int array -> arr
val transpose_conv1d : ?padding:Owl_types_common.padding -> arr -> arr -> int array -> arr
val transpose_conv2d : ?padding:Owl_types_common.padding -> arr -> arr -> int array -> arr
val transpose_conv3d : ?padding:Owl_types_common.padding -> arr -> arr -> int array -> arr
val max_pool1d : ?padding:Owl_types_common.padding -> arr -> int array -> int array -> arr
val max_pool2d : ?padding:Owl_types_common.padding -> arr -> int array -> int array -> arr
val max_pool3d : ?padding:Owl_types_common.padding -> arr -> int array -> int array -> arr
val avg_pool1d : ?padding:Owl_types_common.padding -> arr -> int array -> int array -> arr
val avg_pool2d : ?padding:Owl_types_common.padding -> arr -> int array -> int array -> arr
val avg_pool3d : ?padding:Owl_types_common.padding -> arr -> int array -> int array -> arr
val upsampling2d : arr -> int array -> arr
val conv1d_backward_input : arr -> arr -> int array -> arr -> arr
val conv1d_backward_kernel : arr -> arr -> int array -> arr -> arr
val conv2d_backward_input : arr -> arr -> int array -> arr -> arr
val conv2d_backward_kernel : arr -> arr -> int array -> arr -> arr
val conv3d_backward_input : arr -> arr -> int array -> arr -> arr
val conv3d_backward_kernel : arr -> arr -> int array -> arr -> arr
val dilated_conv1d_backward_input : arr -> arr -> int array -> int array -> arr -> arr
val dilated_conv1d_backward_kernel : arr -> arr -> int array -> int array -> arr -> arr
val dilated_conv2d_backward_input : arr -> arr -> int array -> int array -> arr -> arr
val dilated_conv2d_backward_kernel : arr -> arr -> int array -> int array -> arr -> arr
val dilated_conv3d_backward_input : arr -> arr -> int array -> int array -> arr -> arr
val dilated_conv3d_backward_kernel : arr -> arr -> int array -> int array -> arr -> arr
val transpose_conv1d_backward_input : arr -> arr -> int array -> arr -> arr
val transpose_conv1d_backward_kernel : arr -> arr -> int array -> arr -> arr
val transpose_conv2d_backward_input : arr -> arr -> int array -> arr -> arr
val transpose_conv2d_backward_kernel : arr -> arr -> int array -> arr -> arr
val transpose_conv3d_backward_input : arr -> arr -> int array -> arr -> arr
val transpose_conv3d_backward_kernel : arr -> arr -> int array -> arr -> arr
val max_pool1d_backward : Owl_types_common.padding -> arr -> int array -> int array -> arr -> arr
val max_pool2d_backward : Owl_types_common.padding -> arr -> int array -> int array -> arr -> arr
val max_pool3d_backward : Owl_types_common.padding -> arr -> int array -> int array -> arr -> arr
val avg_pool1d_backward : Owl_types_common.padding -> arr -> int array -> int array -> arr -> arr
val avg_pool2d_backward : Owl_types_common.padding -> arr -> int array -> int array -> arr -> arr
val avg_pool3d_backward : Owl_types_common.padding -> arr -> int array -> int array -> arr -> arr
val upsampling2d_backward : arr -> int array -> arr -> arr
val max_pool2d_argmax : ?padding:Owl_types.padding -> arr -> int array -> int array -> arr * (int64, Stdlib.Bigarray.int64_elt, Stdlib.Bigarray.c_layout) Stdlib.Bigarray.Genarray.t
val conv1d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> unit
val conv2d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> unit
val conv3d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> unit
val dilated_conv1d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> int array -> unit
val dilated_conv2d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> int array -> unit
val dilated_conv3d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> int array -> unit
val transpose_conv1d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> unit
val transpose_conv2d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> unit
val transpose_conv3d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> unit
val max_pool1d_ : out:arr -> ?padding:Owl_types.padding -> arr -> int array -> int array -> unit
val max_pool2d_ : out:arr -> ?padding:Owl_types.padding -> arr -> int array -> int array -> unit
val max_pool3d_ : out:arr -> ?padding:Owl_types.padding -> arr -> int array -> int array -> unit
val avg_pool1d_ : out:arr -> ?padding:Owl_types.padding -> arr -> int array -> int array -> unit
val avg_pool2d_ : out:arr -> ?padding:Owl_types.padding -> arr -> int array -> int array -> unit
val avg_pool3d_ : out:arr -> ?padding:Owl_types.padding -> arr -> int array -> int array -> unit
val upsampling2d_ : out:arr -> arr -> int array -> unit
val conv1d_backward_input_ : out:arr -> arr -> arr -> int array -> arr -> unit
val conv1d_backward_kernel_ : out:arr -> arr -> arr -> int array -> arr -> unit
val conv2d_backward_input_ : out:arr -> arr -> arr -> int array -> arr -> unit
val conv2d_backward_kernel_ : out:arr -> arr -> arr -> int array -> arr -> unit
val conv3d_backward_input_ : out:arr -> arr -> arr -> int array -> arr -> unit
val conv3d_backward_kernel_ : out:arr -> arr -> arr -> int array -> arr -> unit
val dilated_conv1d_backward_input_ : out:arr -> arr -> arr -> int array -> int array -> arr -> unit
val dilated_conv1d_backward_kernel_ : out:arr -> arr -> arr -> int array -> int array -> arr -> unit
val dilated_conv2d_backward_input_ : out:arr -> arr -> arr -> int array -> int array -> arr -> unit
val dilated_conv2d_backward_kernel_ : out:arr -> arr -> arr -> int array -> int array -> arr -> unit
val dilated_conv3d_backward_input_ : out:arr -> arr -> arr -> int array -> int array -> arr -> unit
val dilated_conv3d_backward_kernel_ : out:arr -> arr -> arr -> int array -> int array -> arr -> unit
val transpose_conv1d_backward_input_ : out:arr -> arr -> arr -> int array -> arr -> unit
val transpose_conv1d_backward_kernel_ : out:arr -> arr -> arr -> int array -> arr -> unit
val transpose_conv2d_backward_input_ : out:arr -> arr -> arr -> int array -> arr -> unit
val transpose_conv2d_backward_kernel_ : out:arr -> arr -> arr -> int array -> arr -> unit
val transpose_conv3d_backward_input_ : out:arr -> arr -> arr -> int array -> arr -> unit
val transpose_conv3d_backward_kernel_ : out:arr -> arr -> arr -> int array -> arr -> unit
val max_pool1d_backward_ : out:arr -> Owl_types.padding -> arr -> int array -> int array -> arr -> unit
val max_pool2d_backward_ : out:arr -> Owl_types.padding -> arr -> int array -> int array -> arr -> unit
val max_pool3d_backward_ : out:arr -> Owl_types.padding -> arr -> int array -> int array -> arr -> unit
val avg_pool1d_backward_ : out:arr -> Owl_types.padding -> arr -> int array -> int array -> arr -> unit
val avg_pool2d_backward_ : out:arr -> Owl_types.padding -> arr -> int array -> int array -> arr -> unit
val avg_pool3d_backward_ : out:arr -> Owl_types.padding -> arr -> int array -> int array -> arr -> unit
val upsampling2d_backward_ : out:arr -> arr -> int array -> arr -> unit