package owl

  1. Overview
  2. Docs
Legend:
Library
Module
Module type
Parameter
Class
Class type
type data_format =
  1. | NHWC
  2. | NCHW
type padding =
  1. | SAME
  2. | VALID
val reshape_kernel_2D : ('a, 'b) Owl_dense_ndarray_generic.t -> ('a, 'b) Owl_dense_matrix_generic.t
val pad2d : ('a, 'b) Owl_dense_ndarray_generic.t -> ('c, 'd) Owl_dense_ndarray_generic.t -> int array -> ('a, 'b) Owl_dense_ndarray_generic.t
val owl_conv2d : ?format:data_format -> ?padding:bool -> ('a, 'b) Owl_dense_ndarray_generic.t -> ('a, 'b) Owl_dense_ndarray_generic.t -> int array -> ('a, 'b) Owl_dense_ndarray_generic.t
val im2row : 'a -> 'b option
val calc_conv2d_output_shape : padding -> int -> int -> int -> int -> int -> int -> int * int
val conv2d : ?padding:padding -> ('a, 'b) Owl_dense_ndarray_generic.t -> ('a, 'b) Owl_dense_ndarray_generic.t -> int array -> ('a, 'b) Owl_dense_ndarray_generic.t
val conv2d_backward_input : ('a, 'b) Owl_dense_ndarray_generic.t -> ('a, 'b) Owl_dense_ndarray_generic.t -> int array -> ('a, 'b) Owl_dense_ndarray_generic.t -> ('a, 'b) Owl_dense_ndarray_generic.t
val conv2d_backward_kernel : ('a, 'b) Owl_dense_ndarray_generic.t -> ('a, 'b) Owl_dense_ndarray_generic.t -> int array -> ('a, 'b) Owl_dense_ndarray_generic.t -> ('a, 'b) Owl_dense_ndarray_generic.t
val calc_conv3d_output_shape : padding -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int * int * int
val conv3d : ?padding:padding -> ('a, 'b) Owl_dense_ndarray_generic.t -> ('a, 'b) Owl_dense_ndarray_generic.t -> int array -> ('a, 'b) Owl_dense_ndarray_generic.t
val conv3d_backward_input : ('a, 'b) Owl_dense_ndarray_generic.t -> ('a, 'b) Owl_dense_ndarray_generic.t -> int array -> ('a, 'b) Owl_dense_ndarray_generic.t -> ('a, 'b) Owl_dense_ndarray_generic.t
val conv3d_backward_kernel : ('a, 'b) Owl_dense_ndarray_generic.t -> ('a, 'b) Owl_dense_ndarray_generic.t -> int array -> ('a, 'b) Owl_dense_ndarray_generic.t -> ('a, 'b) Owl_dense_ndarray_generic.t
val algodiff_conv2d : ?padding:padding -> ('a, 'b, Bigarray.c_layout) Bigarray.Array2.t -> ('a, 'b, Bigarray.c_layout) Bigarray.Array2.t -> int array -> int array -> int array -> ('a, 'b, Bigarray.c_layout) Bigarray.Array2.t
OCaml

Innovation. Community. Security.