Module Owl_symbolic_shape
val infer_shape_00 : 'a -> 'b array option arrayval infer_shape_01 : 'a array option array array -> 'a array option arrayval infer_shape_03 : int array option array array -> int array option arrayval infer_shape_05 : int array option array array -> int array -> int array option arrayval infer_shape_07 : int array option array array -> int -> int array option arrayval infer_shape_08 : int array option array array -> int -> int array -> int array option arrayval infer_shape_10 : int array option array array -> int array -> bool -> int array option arrayval infer_shape_11 : int array option array array -> Owl_types.padding -> int array -> int array option arrayval infer_shape_12 : int array option array array -> Owl_types.padding -> int array -> int array option arrayval infer_shape_13 : int array option array array -> Owl_types.padding -> int array -> int array option arrayval infer_shape_14 : int array option array array -> Owl_types.padding -> int array -> int array option arrayval infer_shape_15 : int array option array array -> Owl_types.padding -> int array -> int array -> int array option arrayval infer_shape_17 : int array option array array -> Owl_types.padding -> int array -> int array -> int array option arrayval infer_shape_19 : 'a array option array array -> 'a array option arrayval infer_shape_20 : int array option array array -> int list list -> int array option arrayval infer_shape_21 : int array option array array -> Owl_types.padding -> int array -> int array -> int array option arrayval infer_shape_24 : int array option array array -> Owl_types.padding -> int array -> int array option arrayval infer_shape_25 : int array option array array -> Owl_types.padding -> int array -> int array option arrayval infer_shape_31 : int array option array array -> int array option arrayval infer_shape_32 : 'a array option array array -> int array option arrayval infer_shape_33 : 'a option array array -> 'b array option arrayval infer_shape_34 : int array option array array -> int array option arrayval infer_shape_35 : int array option array array -> int array option arrayval infer_shape_gemm : Owl_symbolic_ops_math.Gemm.t -> 'a array option array array -> 'a array option arrayval infer_shape_pad : Owl_symbolic_ops_tensor.Pad.t -> int array option array array -> int array option arrayval infer_shape_transpose : 'a array option array array -> Owl_symbolic_ops_tensor.Transpose.t -> 'a array option arrayval infer_shape_slice : int array option array array -> Owl_symbolic_ops_tensor.Slice.t -> int array option arrayval infer_shape_space_to_depth : int array option array array -> int -> int array option arrayval infer_shape_conv : int array option array array -> Owl_symbolic_ops_nn.Conv.t -> int array option arrayval infer_shape_conv_transpose : int array option array array -> Owl_symbolic_ops_nn.ConvTranspose.t -> int array option arrayval infer_shape_pool : typ:[< `avg | `max ] -> int array option array array -> int array -> int array -> string -> 'a -> int array option arrayTODO: currently the pads value is not used
val infer_shape_batch_normalization : 'a array option array array -> 'a array option arrayval infer_shape_squeeze : int array option array array -> int array option -> int array option arrayval infer_shape_scatter_elements : 'a array option array array -> 'a array option arrayval infer_shape_scatter_nd : int array option array array -> int array option arrayval infer_shape_gather_elements : 'a array option array array -> 'a array option arrayval infer_shape_instance_norm : 'a array option array array -> 'a array option arrayval infer_shape_flatten : int array option array array -> int -> int array option arrayval infer_shape_lstm : int array option array array -> int array option arrayval infer_shape_roialign : int array option array array -> Owl_symbolic_ops_object_detection.RoiAlign.t -> int array option array
val infer_shape : int array option array array -> Owl_symbolic_symbol.t -> int array option array