NTDSL.Oinclude module type of struct include Ocannl_tensor.Operation.DSL_modules.NTDSL.O endval (*) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval (*.) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval (+) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval threefry4x32 :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval uint4x32_to_prec_uniform :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval uint4x32_to_prec_uniform1 :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval (**.) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Base.float ->
Ocannl_tensor.Tensor.tval relu : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval stretch : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval sat01 : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval fma :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval (!.) : Base.float -> Ocannl_tensor.Tensor.tval (!..) : ?label:??? -> int -> Ocannl_tensor.Tensor.tval (!%) : ?label:??? -> Base.int64 -> Ocannl_tensor.Tensor.tval (!@) : ?label:??? -> Ir.Indexing.static_symbol -> Ocannl_tensor.Tensor.tval dim :
?label:??? ->
Ocannl_tensor.Shape.delayed_var_ref ->
Ocannl_tensor.Tensor.tval (-) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval (~-) : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval (/.) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval (@|) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Operation.Idx.static_symbol ->
Ocannl_tensor.Tensor.tval exp : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval log : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval log2 : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval sin : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval cos : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval neg : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval sqrt : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval stop_gradient :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval recip : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval recip_sqrt : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval tanh : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval where :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval not : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval (<) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval (<=) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval (>) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval (>=) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval (=) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval (<>) :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval embed_self_id : ?label:??? -> unit -> Ocannl_tensor.Tensor.tval einsum :
?label:??? ->
?capture_dims:??? ->
Base.string ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval outer_sum :
?label:??? ->
?capture_dims:??? ->
Base.string ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval einsum1 :
?label:??? ->
?capture_dims:??? ->
Base.string ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval einmax1 :
?label:??? ->
?capture_dims:??? ->
?nonoverlapping:??? ->
Base.string ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval tropical :
?label:??? ->
?capture_dims:??? ->
?nonoverlapping:??? ->
Base.string ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval concat_sum :
?label:??? ->
?capture_dims:??? ->
?negated:??? ->
Base.string ->
Ocannl_tensor.Tensor.t Base.Array.t ->
Ocannl_tensor.Tensor.tval concat :
?label:??? ->
?capture_dims:??? ->
?negated:??? ->
Base.string ->
Ocannl_tensor.Tensor.t Base.Array.t ->
Ocannl_tensor.Tensor.tval stack :
?label:??? ->
Ocannl_tensor.Operation.stack_axis ->
Ocannl_tensor.Tensor.t Base.Array.t ->
Ocannl_tensor.Tensor.tval offsets : ?label:??? -> unit -> Ocannl_tensor.Tensor.tval uniform : ?label:??? -> unit -> Ocannl_tensor.Tensor.tval uniform_at : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval uniform1 : ?label:??? -> unit -> Ocannl_tensor.Tensor.tval uniform_at1 :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval interleave :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval deinterleave_even :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval deinterleave_odd :
?label:??? ->
Ocannl_tensor.Tensor.t ->
Ocannl_tensor.Tensor.tval normal : ?label:??? -> unit -> Ocannl_tensor.Tensor.tval normal1 : ?label:??? -> unit -> Ocannl_tensor.Tensor.tval normal_at : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval normal_at1 : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.tval kaiming :
?label:??? ->
?scale_sq:??? ->
(unit -> Ocannl_tensor.Tensor.t) ->
unit ->
Ocannl_tensor.Tensor.tval xavier :
?label:??? ->
?scale_sq:??? ->
(unit -> Ocannl_tensor.Tensor.t) ->
unit ->
Ocannl_tensor.Tensor.tval kaiming_at :
?label:??? ->
?scale_sq:??? ->
('a -> Ocannl_tensor.Tensor.t) ->
'a ->
Ocannl_tensor.Tensor.tval xavier_at :
?label:??? ->
?scale_sq:??? ->
('a -> Ocannl_tensor.Tensor.t) ->
'a ->
Ocannl_tensor.Tensor.t