DSL_modules.TDSLinclude module type of struct include Ocannl_tensor.Operation.DSL_modules.TDSL endval grad_spec : Ocannl_tensor.Tensor.grad_specval term :
?init_data:Ir.Assignments.init_data ->
?fetch_op:Ocannl_tensor.Tensor.fetch_op ->
Ocannl_tensor.Tensor.op_funval number :
?label:Base.string Base.list ->
?axis_label:Base.string ->
Base.float ->
Ocannl_tensor.Tensor.tval bits :
?label:Base.string Base.list ->
?axis_label:Base.string ->
Base.int64 ->
Ocannl_tensor.Tensor.tval ndarray : Base.float Base.array -> Ocannl_tensor.Tensor.op_funval 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.tval 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_funval 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_funval embed_self_id : ?label:Base.string list -> unit -> Ocannl_tensor.Tensor.tval 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.refval 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_funval 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_funval 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_funval einsum1 :
?capture_dims:Ocannl_tensor.Shape.delayed_var_ref Base.list ->
Base.string ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval einmax1 :
?capture_dims:Ocannl_tensor.Shape.delayed_var_ref Base.list ->
Base.string ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval 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_funval offsets : Ocannl_tensor.Tensor.op_funval range :
?label:Base.string list ->
?axis_label:Base.string ->
Base__Int.t ->
Ocannl_tensor.Tensor.tval 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.tval stop_gradient : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval 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.tval 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.tval 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.tval 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 ->
?deduced:Ocannl_tensor.Shape.deduce_within_shape ->
Base.unit ->
Ocannl_tensor.Tensor.tval 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.tval 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_funval 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.tval 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.tval matmul :
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval pointmul :
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval add :
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval pointpow :
Base.float ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval relu : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval sat01 : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval number_int :
?label:Base.string Base.list ->
?axis_label:Base.string ->
int ->
Ocannl_tensor.Tensor.tval embed_symbol :
?label:Base.string list ->
Ir.Indexing.static_symbol ->
Ocannl_tensor.Tensor.tval embed_dim :
?label:Base.string list ->
Ocannl_tensor.Shape.delayed_var_ref ->
Ocannl_tensor.Tensor.tval sub :
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval pointdiv :
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval exp : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval log : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval log2 : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval sin : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval cos : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval neg : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval sqrt : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval recip : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval recip_sqrt : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval tanh : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval where :
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval not : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval 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_funval 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_funval 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_funval interleave :
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval normal : unit -> Ocannl_tensor.Tensor.op_funval normal1 : unit -> Ocannl_tensor.Tensor.op_funval normal_at : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval normal_at1 : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funmodule O : sig ... end