NN (owl-base.Owl_base_dense_ndarray

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