owl-symbolic
  1. Overview
  2. Docs
package owl-symbolic
Legend:
Library
Module
Module type
Parameter
Class
Class type
val infer_shape_00 : 'a -> 'b array option array
val infer_shape_01 : 'a array option array array -> 'a array option array
val infer_shape_03 : int array option array array -> int array option array
val infer_shape_05 : int array option array array -> int array -> int array option array
val infer_shape_07 : int array option array array -> int -> int array option array
val infer_shape_08 : int array option array array -> int -> int array -> int array option array
val infer_shape_10 : int array option array array -> int array -> bool -> int array option array
val infer_shape_11 : int array option array array -> Owl_types.padding -> int array -> int array option array
val infer_shape_12 : int array option array array -> Owl_types.padding -> int array -> int array option array
val infer_shape_13 : int array option array array -> Owl_types.padding -> int array -> int array option array
val infer_shape_14 : int array option array array -> Owl_types.padding -> int array -> int array option array
val infer_shape_15 : int array option array array -> Owl_types.padding -> int array -> int array -> int array option array
val infer_shape_17 : int array option array array -> Owl_types.padding -> int array -> int array -> int array option array
val infer_shape_19 : 'a array option array array -> 'a array option array
val infer_shape_20 : int array option array array -> int list list -> int array option array
val infer_shape_21 : int array option array array -> Owl_types.padding -> int array -> int array -> int array option array
val infer_shape_24 : int array option array array -> Owl_types.padding -> int array -> int array option array
val infer_shape_25 : int array option array array -> Owl_types.padding -> int array -> int array option array
val infer_shape_31 : int array option array array -> int array option array
val infer_shape_32 : 'a array option array array -> int array option array
val infer_shape_33 : 'a option array array -> 'b array option array
val infer_shape_34 : int array option array array -> int array option array
val infer_shape_35 : int array option array array -> int array option array
val infer_shape_det : 'a array option array array -> 'a array option array
val infer_shape_gemm : Owl_symbolic_ops_math.Gemm.t -> 'a array option array array -> 'a array option array
val infer_shape_pad : Owl_symbolic_ops_tensor.Pad.t -> int array option array array -> int array option array
val infer_shape_transpose : 'a array option array array -> Owl_symbolic_ops_tensor.Transpose.t -> 'a array option array
val infer_shape_slice : int array option array array -> Owl_symbolic_ops_tensor.Slice.t -> int array option array
val infer_shape_space_to_depth : int array option array array -> int -> int array option array
val infer_shape_depth_to_space : int array option array array -> int -> int array option array
val infer_shape_conv : int array option array array -> Owl_symbolic_ops_nn.Conv.t -> int array option array
val infer_shape_conv_transpose : int array option array array -> Owl_symbolic_ops_nn.ConvTranspose.t -> int array option array
val infer_shape_pool : typ:[< `avg | `max ] -> int array option array array -> int array -> int array -> string -> 'a -> int array option array

TODO: currently the pads value is not used

val infer_shape_batch_normalization : 'a array option array array -> 'a array option array
val infer_shape_squeeze : int array option array array -> int array option -> int array option array
val infer_shape_unsqueeze : int array option array array -> int array -> int array option array
val infer_shape_scatter_elements : 'a array option array array -> 'a array option array
val infer_shape_scatter_nd : int array option array array -> int array option array
val infer_shape_gather_elements : 'a array option array array -> 'a array option array
val infer_shape_instance_norm : 'a array option array array -> 'a array option array
val infer_shape_flatten : int array option array array -> int -> int array option array
val infer_shape_lstm : int array option array array -> int array option array
val infer_shape_rnn : int array option array array -> int array option array
val infer_shape_roialign : int array option array array -> Owl_symbolic_ops_object_detection.RoiAlign.t -> int array option array
val infer_shape_axis : 'a array option array array -> int -> 'a array option array
val infer_shape_ql_matmul : 'a array option array array -> 'a array option array
val infer_shape_compress : 'a option array array -> 'b option -> 'a option array
val infer_shape_revseq : 'a array option array array -> 'a array option array
val infer_shape_seq_at : 'a array array -> int -> 'a array
val infer_shape_seq_cons : 'a array array -> 'a array
val infer_shape_seq_insert : 'a array array -> int -> 'a array
val infer_shape_seq_erase : 'a array array -> int -> 'a array
val infer_shape_split_to_seq : int array option array array -> Owl_symbolic_ops_sequence.SplitToSequence.t -> int array option array
val infer_shape_concat_from_seq : int array option array array -> int -> bool -> int array option array
val infer_shape_non_max_suppression : int array option array array -> 'a option array
val infer_shape_resize : int array option array array -> float array option -> int array option -> int array option array
val infer_shape_onehot : 'a array option array array -> int -> 'a -> 'a array option array

Main entry

val infer_shape_eye_like : 'a array option array array -> 'a array option array
val infer_shape_multinomial : 'a array option array array -> 'a -> 'a array option array
val infer_shape_dynamic_quantize : 'a array option array array -> 'a array option array
val infer_shape : int array option array array -> Owl_symbolic_symbol.t -> int array option array