PDSL (neural_nets_lib.Ocannl.Nn_blocks.DSL

include module type of struct include Ocannl_tensor.Operation.DSL_modules.PDSL end

val grad_spec : Ocannl_tensor.Tensor.grad_spec

val term : 
  ?init_data:Ir.Assignments.init_data ->
  ?fetch_op:Ocannl_tensor.Tensor.fetch_op ->
  Ocannl_tensor.Tensor.op_fun

val number : 
  ?label:Base.string Base.list ->
  ?axis_label:Base.string ->
  Base.float ->
  Ocannl_tensor.Tensor.t

val bits : 
  ?label:Base.string Base.list ->
  ?axis_label:Base.string ->
  Base.int64 ->
  Ocannl_tensor.Tensor.t

val ndarray : Base.float Base.array -> Ocannl_tensor.Tensor.op_fun

val threefry4x32 : 
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  ?label:Base.string Base.list ->
  ?batch_dims:Base.int Base.list ->
  ?batch_axes:(Base.string * Base.int) Base.list ->
  ?input_dims:Base.int Base.list ->
  ?output_dims:Base.int Base.list ->
  ?input_axes:(Base.string * Base.int) Base.list ->
  ?output_axes:(Base.string * Base.int) Base.list ->
  ?deduced:Ocannl_tensor.Shape.deduce_within_shape ->
  unit ->
  Ocannl_tensor.Tensor.t

val uint4x32_to_prec_uniform : 
  Ocannl_tensor.Tensor.t ->
  ?label:Base.string Base.list ->
  ?top_down_prec:Base.bool ->
  ?batch_dims:Base.int Base.list ->
  ?batch_axes:(Base.string * Base.int) Base.list ->
  Ocannl_tensor.Tensor.param_op_fun

val uint4x32_to_prec_uniform1 : 
  Ocannl_tensor.Tensor.t ->
  ?label:Base.string Base.list ->
  ?top_down_prec:Base.bool ->
  ?batch_dims:Base.int Base.list ->
  ?batch_axes:(Base.string * Base.int) Base.list ->
  Ocannl_tensor.Tensor.param_op_fun

val embed_self_id : ?label:Base.string list -> unit -> Ocannl_tensor.Tensor.t

val default_param_init : 
  (unit ->
    ?label:Base.string Base.list ->
    ?top_down_prec:Base.bool ->
    ?batch_dims:Base.int Base.list ->
    ?batch_axes:(Base.string * Base.int) Base.list ->
    Ocannl_tensor.Tensor.param_op_fun)
    Base.ref

val param : 
  ?value:Base.float ->
  ?values:Base.float Base.array ->
  ?param_init:Ocannl_tensor.Tensor.op_fun ->
  Base.string ->
  ?more_label:Base.string Base.list ->
  Ocannl_tensor.Tensor.param_op_fun

val einsum : 
  ?capture_dims:Ocannl_tensor.Shape.delayed_var_ref Base.list ->
  Base.string ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val outer_sum : 
  ?capture_dims:Ocannl_tensor.Shape.delayed_var_ref Base.list ->
  Base.string ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val einsum1 : 
  ?capture_dims:Ocannl_tensor.Shape.delayed_var_ref Base.list ->
  Base.string ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val einmax1 : 
  ?capture_dims:Ocannl_tensor.Shape.delayed_var_ref Base.list ->
  Base.string ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val tropical : 
  ?capture_dims:Ocannl_tensor.Shape.delayed_var_ref Base.list ->
  Base.string ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val offsets : Ocannl_tensor.Tensor.op_fun

val range : 
  ?label:Base.string list ->
  ?axis_label:Base.string ->
  Base__Int.t ->
  Ocannl_tensor.Tensor.t

val range_of_shape : 
  ?label:Base.string list ->
  ?batch_dims:Base.Int.t Base.List.t ->
  ?input_dims:Base.Int.t Base.List.t ->
  ?output_dims:Base.Int.t Base.List.t ->
  ?batch_axes:(Base.string * Base.Int.t) Base.List.t ->
  ?input_axes:(Base.string * Base.Int.t) Base.List.t ->
  ?output_axes:(Base.string * Base.Int.t) Base.List.t ->
  unit ->
  Ocannl_tensor.Tensor.t

val stop_gradient : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val reshape : 
  l:Base.string ->
  ?b:Base.int Base.list ->
  ?i:Base.int Base.list ->
  ?o:Base.int Base.list ->
  Ir.Ndarray.t ->
  ?fetch_op:Ocannl_tensor.Tensor.fetch_op ->
  ?top_down_prec:Base.bool ->
  ?batch_axes:(Base.string * Base.int) Base.list ->
  ?input_axes:(Base.string * Base.int) Base.list ->
  ?output_axes:(Base.string * Base.int) Base.list ->
  ?deduced:Ocannl_tensor.Shape.deduce_within_shape ->
  Base.unit ->
  Ocannl_tensor.Tensor.t

val wrap : 
  l:Base.string ->
  ?b:Base.int Base.list ->
  ?i:Base.int Base.list ->
  ?o:Base.int Base.list ->
  Ir.Ndarray.t ->
  ?fetch_op:Ocannl_tensor.Tensor.fetch_op ->
  ?top_down_prec:Base.bool ->
  ?batch_axes:(Base.string * Base.int) Base.list ->
  ?input_axes:(Base.string * Base.int) Base.list ->
  ?output_axes:(Base.string * Base.int) Base.list ->
  ?deduced:Ocannl_tensor.Shape.deduce_within_shape ->
  Base.unit ->
  Ocannl_tensor.Tensor.t

val wrap_padded : 
  l:Base.string ->
  ?b:Base.int Base.list ->
  ?i:Base.int Base.list ->
  ?o:Base.int Base.list ->
  padding:Ir.Ops.axis_padding Base.array ->
  padded_value:Base.float ->
  Ir.Assignments.Nd.t ->
  ?fetch_op:Ocannl_tensor.Tensor.fetch_op ->
  ?top_down_prec:Base.bool ->
  ?batch_axes:(Base.string * Base.int) Base.list ->
  ?input_axes:(Base.string * Base.int) Base.list ->
  ?output_axes:(Base.string * Base.int) Base.list ->
  ?deduced:Ocannl_tensor.Shape.deduce_within_shape ->
  Base.unit ->
  Ocannl_tensor.Tensor.t

val rebatch : 
  l:Base.string ->
  Ir.Ndarray.t ->
  ?fetch_op:Ocannl_tensor.Tensor.fetch_op ->
  ?top_down_prec:Base.bool ->
  ?batch_dims:Base.int Base.list ->
  ?batch_axes:(Base.string * Base.int) Base.list ->
  ?input_axes:(Base.string * Base.int) Base.list ->
  ?output_axes:(Base.string * Base.int) Base.list ->
  ?deduced:Ocannl_tensor.Shape.deduce_within_shape ->
  Base.unit ->
  Ocannl_tensor.Tensor.t

val init : 
  l:Base.string ->
  prec:Ir.Ops.prec ->
  ?b:Base.int Base__List.t ->
  ?i:Base.int Base__List.t ->
  ?o:Base.int Base__List.t ->
  f:(Base.int Base.array -> Base.float) ->
  ?fetch_op:Ocannl_tensor.Tensor.fetch_op ->
  ?top_down_prec:Base.bool ->
  ?batch_axes:(Base.string * Base.int) Base.list ->
  ?input_axes:(Base.string * Base.int) Base.list ->
  ?output_axes:(Base.string * Base.int) Base.list ->
  ?deduced:Ocannl_tensor.Shape.deduce_within_shape ->
  Base.unit ->
  Ocannl_tensor.Tensor.t

val uniform : 
  unit ->
  ?label:Base.string Base.list ->
  ?top_down_prec:Base.bool ->
  ?batch_dims:Base.int Base.list ->
  ?batch_axes:(Base.string * Base.int) Base.list ->
  Ocannl_tensor.Tensor.param_op_fun

val reshape_param : 
  l:Base.string ->
  ?i:Base.int Base.list ->
  ?o:Base.int Base.list ->
  Ir.Ndarray.t ->
  ?more_label:Base.string Base.list ->
  ?input_axes:(Base.string * Base.int) Base.list ->
  ?output_axes:(Base.string * Base.int) Base.list ->
  ?deduced:Ocannl_tensor.Shape.deduce_within_shape ->
  Base.unit ->
  Ocannl_tensor.Tensor.t

val wrap_param : 
  l:Base.string ->
  ?i:Base.int Base.list ->
  ?o:Base.int Base.list ->
  Ir.Ndarray.t ->
  ?more_label:Base.string Base.list ->
  ?input_axes:(Base.string * Base.int) Base.list ->
  ?output_axes:(Base.string * Base.int) Base.list ->
  ?deduced:Ocannl_tensor.Shape.deduce_within_shape ->
  Base.unit ->
  Ocannl_tensor.Tensor.t

val matmul : 
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val pointmul : 
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val add : 
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val pointpow : 
  Base.float ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val relu : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val sat01 : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val fma : 
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val number_int : 
  ?label:Base.string Base.list ->
  ?axis_label:Base.string ->
  int ->
  Ocannl_tensor.Tensor.t

val embed_symbol : 
  ?label:Base.string list ->
  Ir.Indexing.static_symbol ->
  Ocannl_tensor.Tensor.t

val embed_dim : 
  ?label:Base.string list ->
  Ocannl_tensor.Shape.delayed_var_ref ->
  Ocannl_tensor.Tensor.t

val sub : 
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val pointdiv : 
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val slice : 
  Ocannl_tensor.Operation.Idx.static_symbol ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val exp : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val log : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val log2 : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val sin : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val cos : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val neg : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val sqrt : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val recip : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val recip_sqrt : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val tanh : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val where : 
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val not : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val lt : 
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val eq : 
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val ne : 
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.t ->
  Ocannl_tensor.Tensor.op_fun

val uniform_at : 
  Ocannl_tensor.Tensor.t ->
  ?label:Base.string Base.list ->
  ?top_down_prec:Base.bool ->
  ?batch_dims:Base.int Base.list ->
  ?batch_axes:(Base.string * Base.int) Base.list ->
  Ocannl_tensor.Tensor.param_op_fun

val uniform1 : 
  unit ->
  ?label:Base.string Base.list ->
  ?top_down_prec:Base.bool ->
  ?batch_dims:Base.int Base.list ->
  ?batch_axes:(Base.string * Base.int) Base.list ->
  Ocannl_tensor.Tensor.param_op_fun

val uniform_at1 : 
  Ocannl_tensor.Tensor.t ->
  ?label:Base.string Base.list ->
  ?top_down_prec:Base.bool ->
  ?batch_dims:Base.int Base.list ->
  ?batch_axes:(Base.string * Base.int) Base.list ->
  Ocannl_tensor.Tensor.param_op_fun

val normal : unit -> Ocannl_tensor.Tensor.op_fun

val normal1 : unit -> Ocannl_tensor.Tensor.op_fun

val normal_at : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

val normal_at1 : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_fun

module O : sig ... end