Module TDSL.O

include module type of struct include Ocannl_tensor.Operation.DSL_modules.TDSL.O end
val uint4x32_to_prec_uniform : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val uint4x32_to_prec_uniform1 : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val (**.) : ?label:??? -> Ocannl_tensor.Tensor.t -> Base.float -> Ocannl_tensor.Tensor.t
val stretch : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val sat01 : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val (!.) : Base.float -> Ocannl_tensor.Tensor.t
val (!..) : ?label:??? -> int -> Ocannl_tensor.Tensor.t
val (!%) : ?label:??? -> Base.int64 -> Ocannl_tensor.Tensor.t
val stop_gradient : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val recip : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val recip_sqrt : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val embed_self_id : ?label:??? -> unit -> Ocannl_tensor.Tensor.t
val einsum : ?label:??? -> ?capture_dims:??? -> Base.string -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val outer_sum : ?label:??? -> ?capture_dims:??? -> Base.string -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val einsum1 : ?label:??? -> ?capture_dims:??? -> Base.string -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val einmax1 : ?label:??? -> ?capture_dims:??? -> ?nonoverlapping:??? -> Base.string -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val tropical : ?label:??? -> ?capture_dims:??? -> ?nonoverlapping:??? -> Base.string -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val concat_sum : ?label:??? -> ?capture_dims:??? -> ?negated:??? -> Base.string -> Ocannl_tensor.Tensor.t Base.Array.t -> Ocannl_tensor.Tensor.t
val concat : ?label:??? -> ?capture_dims:??? -> ?negated:??? -> Base.string -> Ocannl_tensor.Tensor.t Base.Array.t -> Ocannl_tensor.Tensor.t
val offsets : ?label:??? -> unit -> Ocannl_tensor.Tensor.t
val uniform : ?label:??? -> unit -> Ocannl_tensor.Tensor.t
val uniform_at : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val uniform1 : ?label:??? -> unit -> Ocannl_tensor.Tensor.t
val uniform_at1 : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val deinterleave_even : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val deinterleave_odd : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val normal : ?label:??? -> unit -> Ocannl_tensor.Tensor.t
val normal1 : ?label:??? -> unit -> Ocannl_tensor.Tensor.t
val normal_at : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val normal_at1 : ?label:??? -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val kaiming : ?label:??? -> ?scale_sq:??? -> (unit -> Ocannl_tensor.Tensor.t) -> unit -> Ocannl_tensor.Tensor.t
val xavier : ?label:??? -> ?scale_sq:??? -> (unit -> Ocannl_tensor.Tensor.t) -> unit -> Ocannl_tensor.Tensor.t
val kaiming_at : ?label:??? -> ?scale_sq:??? -> ('a -> Ocannl_tensor.Tensor.t) -> 'a -> Ocannl_tensor.Tensor.t
val xavier_at : ?label:??? -> ?scale_sq:??? -> ('a -> Ocannl_tensor.Tensor.t) -> 'a -> Ocannl_tensor.Tensor.t