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 ) Stdlib.Bigarray.kind -> ( 'a, 'b ) Owl_core_types.owl_arr_op26
val _owl_spatial_avg_pooling : 'a 'b. ( 'a, 'b ) Stdlib.Bigarray.kind -> ( 'a, 'b ) Owl_core_types.owl_arr_op26
val _owl_cuboid_max_pooling : 'a 'b. ( 'a, 'b ) Stdlib.Bigarray.kind -> ( 'a, 'b ) Owl_core_types.owl_arr_op27
val _owl_cuboid_avg_pooling : 'a 'b. ( 'a, 'b ) Stdlib.Bigarray.kind -> ( 'a, 'b ) Owl_core_types.owl_arr_op27
val _owl_spatial_max_pooling_argmax : 'a 'b. ( 'a, 'b ) Stdlib.Bigarray.kind -> ( 'a, 'b ) Owl_core_types.owl_arr_op28
val _owl_spatial_max_pooling_backward : 'a 'b. ( 'a, 'b ) Stdlib.Bigarray.kind -> ( 'a, 'b ) Owl_core_types.owl_arr_op29
val _owl_spatial_avg_pooling_backward : 'a 'b. ( 'a, 'b ) Stdlib.Bigarray.kind -> ( 'a, 'b ) Owl_core_types.owl_arr_op30
val _owl_cuboid_max_pooling_backward : 'a 'b. ( 'a, 'b ) Stdlib.Bigarray.kind -> ( 'a, 'b ) Owl_core_types.owl_arr_op31
val _owl_cuboid_avg_pooling_backward : 'a 'b. ( 'a, 'b ) Stdlib.Bigarray.kind -> ( 'a, 'b ) Owl_core_types.owl_arr_op32