DSL_modules.NTDSLinclude module type of struct include Ocannl_tensor.Operation.DSL_modules.NTDSL endval grad_spec : Ocannl_tensor.Tensor.grad_specval term : ?init_data:??? -> ?fetch_op:??? -> Ocannl_tensor.Tensor.op_funval number :
?label:??? ->
?axis_basis:??? ->
Base.float ->
Ocannl_tensor.Tensor.tval bits :
?label:??? ->
?axis_basis:??? ->
Base.int64 ->
Ocannl_tensor.Tensor.tval ndarray : Base.float Base.array -> Ocannl_tensor.Tensor.op_funval threefry4x32 :
?spec:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
?label:??? ->
?batch_dims:??? ->
?batch_axes:??? ->
?input_dims:??? ->
?output_dims:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
unit ->
Ocannl_tensor.Tensor.tval uint4x32_to_prec_uniform :
Ocannl_tensor.Tensor.t ->
?label:??? ->
?top_down_prec:??? ->
?batch_dims:??? ->
?batch_axes:??? ->
?input_dims:??? ->
?output_dims:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
unit ->
Ocannl_tensor.Tensor.tval uint4x32_to_prec_uniform1 :
Ocannl_tensor.Tensor.t ->
?label:??? ->
?top_down_prec:??? ->
?batch_dims:??? ->
?batch_axes:??? ->
?input_dims:??? ->
?output_dims:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
unit ->
Ocannl_tensor.Tensor.tval embed_self_id : ?label:??? -> unit -> Ocannl_tensor.Tensor.tval default_param_init :
(unit ->
?label:??? ->
?top_down_prec:??? ->
?batch_dims:??? ->
?batch_axes:??? ->
?input_dims:??? ->
?output_dims:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
unit ->
Ocannl_tensor.Tensor.t)
Base.refval param :
?value:??? ->
?values:??? ->
?param_init:??? ->
Base.string ->
?more_label:??? ->
Ocannl_tensor.Tensor.param_op_funval einsum :
?capture_dims:??? ->
Base.string ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval outer_sum :
?capture_dims:??? ->
Base.string ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval einsum1 :
?capture_dims:??? ->
Base.string ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval einmax1 :
?capture_dims:??? ->
?nonoverlapping:??? ->
Base.string ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval tropical :
?capture_dims:??? ->
?nonoverlapping:??? ->
Base.string ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval concat_sum :
?capture_dims:??? ->
?negated:??? ->
Base.string ->
Ocannl_tensor.Tensor.t Base.Array.t ->
Ocannl_tensor.Tensor.op_funval concat :
?capture_dims:??? ->
?negated:??? ->
Base.string ->
Ocannl_tensor.Tensor.t Base.Array.t ->
Ocannl_tensor.Tensor.op_funval stack :
Ocannl_tensor.Operation.stack_axis ->
Ocannl_tensor.Tensor.t Base.Array.t ->
Ocannl_tensor.Tensor.op_funval offsets : Ocannl_tensor.Tensor.op_funval range :
?label:??? ->
?axis_basis:??? ->
Base__Int.t ->
Ocannl_tensor.Tensor.tval range_of_shape :
?label:??? ->
?batch_dims:??? ->
?input_dims:??? ->
?output_dims:??? ->
?batch_axes:??? ->
?input_axes:??? ->
?output_axes:??? ->
unit ->
Ocannl_tensor.Tensor.tval stop_gradient :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval reshape :
l:Base.string ->
?b:??? ->
?i:??? ->
?o:??? ->
Ir.Ndarray.t ->
?fetch_op:??? ->
?top_down_prec:??? ->
?batch_axes:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
Base.unit ->
Ocannl_tensor.Tensor.tval wrap :
l:Base.string ->
?prec:??? ->
?b:??? ->
?i:??? ->
?o:??? ->
Ir.Ndarray.t ->
?fetch_op:??? ->
?top_down_prec:??? ->
?batch_axes:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
Base.unit ->
Ocannl_tensor.Tensor.tval wrap_padded :
l:Base.string ->
?b:??? ->
?i:??? ->
?o:??? ->
padding:Ir.Ops.axis_padding Base.array ->
padded_value:Base.float ->
Ir.Ndarray.t ->
?fetch_op:??? ->
?top_down_prec:??? ->
?batch_axes:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
Base.unit ->
Ocannl_tensor.Tensor.tval rebatch :
l:Base.string ->
Ir.Ndarray.t ->
?fetch_op:??? ->
?top_down_prec:??? ->
?batch_dims:??? ->
?batch_axes:??? ->
?deduced:??? ->
Base.unit ->
Ocannl_tensor.Tensor.tval init :
l:Base.string ->
prec:Ir.Ops.prec ->
?b:??? ->
?i:??? ->
?o:??? ->
f:(Base.int Base.array -> Base.float) ->
?fetch_op:??? ->
?top_down_prec:??? ->
?batch_axes:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
Base.unit ->
Ocannl_tensor.Tensor.tval uniform :
unit ->
?label:??? ->
?top_down_prec:??? ->
?batch_dims:??? ->
?batch_axes:??? ->
?input_dims:??? ->
?output_dims:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
unit ->
Ocannl_tensor.Tensor.tval reshape_param :
l:Base.string ->
?i:??? ->
?o:??? ->
Ir.Ndarray.t ->
?more_label:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
Base.unit ->
Ocannl_tensor.Tensor.tval wrap_param :
l:Base.string ->
?i:??? ->
?o:??? ->
Ir.Ndarray.t ->
?more_label:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
Base.unit ->
Ocannl_tensor.Tensor.tval matmul :
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval pointmul :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval add :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval pointpow :
?spec:??? ->
?capture_dims:??? ->
Base.float ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval relu :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval sat01 :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval number_int : ?label:??? -> ?axis_basis:??? -> int -> Ocannl_tensor.Tensor.tval embed_symbol :
?label:??? ->
Ir.Indexing.static_symbol ->
Ocannl_tensor.Tensor.tval embed_dim :
?label:??? ->
Ocannl_tensor.Shape.delayed_var_ref ->
Ocannl_tensor.Tensor.tval sub :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval pointdiv :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval exp :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval log :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval log2 :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval sin :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval cos :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval neg :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval cast :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval stretch :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
?label:??? ->
?top_down_prec:??? ->
?batch_dims:??? ->
?batch_axes:??? ->
?input_dims:??? ->
?output_dims:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
unit ->
Ocannl_tensor.Tensor.tval sqrt :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval recip :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval recip_sqrt :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval tanh :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval where :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval not :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval lt :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval le :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval eq :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval ne :
?spec:??? ->
?capture_dims:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval uniform_at :
Ocannl_tensor.Tensor.t ->
?label:??? ->
?top_down_prec:??? ->
?batch_dims:??? ->
?batch_axes:??? ->
?input_dims:??? ->
?output_dims:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
unit ->
Ocannl_tensor.Tensor.tval uniform1 :
unit ->
?label:??? ->
?top_down_prec:??? ->
?batch_dims:??? ->
?batch_axes:??? ->
?input_dims:??? ->
?output_dims:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
unit ->
Ocannl_tensor.Tensor.tval uniform_at1 :
Ocannl_tensor.Tensor.t ->
?label:??? ->
?top_down_prec:??? ->
?batch_dims:??? ->
?batch_axes:??? ->
?input_dims:??? ->
?output_dims:??? ->
?input_axes:??? ->
?output_axes:??? ->
?deduced:??? ->
unit ->
Ocannl_tensor.Tensor.tval interleave :
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.op_funval deinterleave_even : Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.op_funval deinterleave_odd : 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_funval kaiming :
?scale_sq:??? ->
(unit -> Ocannl_tensor.Tensor.t) ->
unit ->
Ocannl_tensor.Tensor.op_funval xavier :
?scale_sq:??? ->
(unit -> Ocannl_tensor.Tensor.t) ->
unit ->
Ocannl_tensor.Tensor.op_funval kaiming_at :
?scale_sq:??? ->
('a -> Ocannl_tensor.Tensor.t) ->
'a ->
Ocannl_tensor.Tensor.op_funval xavier_at :
?scale_sq:??? ->
('a -> Ocannl_tensor.Tensor.t) ->
'a ->
Ocannl_tensor.Tensor.op_funmodule O : sig ... end