Module Owl_utils_infer_shapeSource

Sourceval require_broadcasting : int array -> int array -> bool
Sourceval calc_conv2d_output_shape : Owl_types.padding -> int -> int -> int -> int -> int -> int -> int * int
Sourceval calc_transpose_conv2d_output_shape : Owl_types.padding -> int -> int -> int -> int -> int -> int -> int * int
Sourceval calc_conv2d_padding : int -> int -> int -> int -> int -> int -> int -> int -> int * int * int * int
Sourceval calc_conv1d_output_shape : Owl_types.padding -> int -> int -> int -> int
Sourceval calc_transpose_conv1d_output_shape : Owl_types.padding -> int -> int -> int -> int
Sourceval calc_conv3d_output_shape : Owl_types.padding -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int * int * int
Sourceval calc_transpose_conv3d_output_shape : Owl_types.padding -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int * int * int
Sourceval calc_conv3d_padding : int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int -> int * int * int * int * int * int
Sourceval broadcast1 : int array -> int array -> int array
Sourceval broadcast2 : int array -> int array -> int array -> int array
Sourceval broadcast1_stride : int array -> int array -> int array * int array
Sourceval fold : int array -> int -> int array
Sourceval tile : int array -> int array -> int array
Sourceval repeat : int array -> int array -> int array
Sourceval concatenate : int array array -> int -> int array
Sourceval split : int array -> int -> int array -> int array array
Sourceval draw : 'a array -> int -> 'a -> 'a array
Sourceval reduce : int array -> int array -> int array
Sourceval conv2d : int array -> Owl_types.padding -> int array -> int array -> int array
Sourceval conv1d : int array -> Owl_types.padding -> int array -> int array -> int array
Sourceval conv3d : int array -> Owl_types.padding -> int array -> int array -> int array
Sourceval dilated_conv2d : int array -> Owl_types.padding -> int array -> int array -> int array -> int array
Sourceval dilated_conv1d : int array -> Owl_types.padding -> int array -> int array -> int array -> int array
Sourceval dilated_conv3d : int array -> Owl_types.padding -> int array -> int array -> int array -> int array
Sourceval transpose_conv2d : int array -> Owl_types.padding -> int array -> int array -> int array
Sourceval transpose_conv1d : int array -> Owl_types.padding -> int array -> int array -> int array
Sourceval transpose_conv3d : int array -> Owl_types.padding -> int array -> int array -> int array
Sourceval pool2d : int array -> Owl_types.padding -> int array -> int array -> int array
Sourceval upsampling2d : int array -> int array -> int array
Sourceval transpose : 'a array -> int array -> 'a array
Sourceval dot : 'a array -> 'a array -> 'a array
Sourceval onehot : 'a array -> 'a -> 'a array