Module Owl_ndarray_pool

val owl_float32_ndarray_maxpool_spatial : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float32_ndarray_avgpool_spatial : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float32_ndarray_maxpool_spatial_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val _owl_float32_ndarray_avgpool_spatial_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float32_ndarray_avgpool_spatial_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float32_ndarray_maxpool_argmax_spatial : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> (int64, 'c) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float32_ndarray_maxpool_cuboid : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float32_ndarray_avgpool_cuboid : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float32_ndarray_maxpool_cuboid_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val _owl_float32_ndarray_avgpool_cuboid_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float32_ndarray_avgpool_cuboid_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float64_ndarray_maxpool_spatial : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float64_ndarray_avgpool_spatial : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float64_ndarray_maxpool_spatial_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val _owl_float64_ndarray_avgpool_spatial_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float64_ndarray_avgpool_spatial_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float64_ndarray_maxpool_cuboid : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float64_ndarray_avgpool_cuboid : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float64_ndarray_maxpool_cuboid_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val _owl_float64_ndarray_avgpool_cuboid_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float64_ndarray_avgpool_cuboid_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_float64_ndarray_maxpool_argmax_spatial : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> (int64, 'c) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex32_ndarray_maxpool_spatial : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex32_ndarray_avgpool_spatial : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex32_ndarray_maxpool_spatial_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val _owl_complex32_ndarray_avgpool_spatial_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex32_ndarray_avgpool_spatial_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex32_ndarray_maxpool_argmax_spatial : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> (int64, 'c) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex32_ndarray_maxpool_cuboid : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex32_ndarray_avgpool_cuboid : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex32_ndarray_maxpool_cuboid_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val _owl_complex32_ndarray_avgpool_cuboid_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex32_ndarray_avgpool_cuboid_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex64_ndarray_maxpool_spatial : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex64_ndarray_avgpool_spatial : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex64_ndarray_maxpool_spatial_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val _owl_complex64_ndarray_avgpool_spatial_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex64_ndarray_avgpool_spatial_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex64_ndarray_maxpool_cuboid : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex64_ndarray_avgpool_cuboid : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex64_ndarray_maxpool_cuboid_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val _owl_complex64_ndarray_avgpool_cuboid_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex64_ndarray_avgpool_cuboid_backward : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val owl_complex64_ndarray_maxpool_argmax_spatial : ('a, 'b) Owl_core_types.owl_arr -> ('a, 'b) Owl_core_types.owl_arr -> (int64, 'c) Owl_core_types.owl_arr -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> unit
val _owl_spatial_max_pooling : 'a 'b. ('a, 'b) Bigarray.kind -> ('a, 'b) Owl_core_types.owl_arr_op26
val _owl_spatial_avg_pooling : 'a 'b. ('a, 'b) Bigarray.kind -> ('a, 'b) Owl_core_types.owl_arr_op26
val _owl_cuboid_max_pooling : 'a 'b. ('a, 'b) Bigarray.kind -> ('a, 'b) Owl_core_types.owl_arr_op27
val _owl_cuboid_avg_pooling : 'a 'b. ('a, 'b) Bigarray.kind -> ('a, 'b) Owl_core_types.owl_arr_op27
val _owl_spatial_max_pooling_argmax : 'a 'b. ('a, 'b) Bigarray.kind -> ('a, 'b) Owl_core_types.owl_arr_op28
val _owl_spatial_max_pooling_backward : 'a 'b. ('a, 'b) Bigarray.kind -> ('a, 'b) Owl_core_types.owl_arr_op29
val _owl_spatial_avg_pooling_backward : 'a 'b. ('a, 'b) Bigarray.kind -> ('a, 'b) Owl_core_types.owl_arr_op30
val _owl_cuboid_max_pooling_backward : 'a 'b. ('a, 'b) Bigarray.kind -> ('a, 'b) Owl_core_types.owl_arr_op31
val _owl_cuboid_avg_pooling_backward : 'a 'b. ('a, 'b) Bigarray.kind -> ('a, 'b) Owl_core_types.owl_arr_op32