Module TDSL.O

include module type of struct include Ocannl_tensor.Operation.DSL_modules.TDSL.O end
val (*) : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val (*.) : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val (+) : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val threefry4x32 : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val uint4x32_to_prec_uniform : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val uint4x32_to_prec_uniform1 : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val (**.) : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Base.float -> Ocannl_tensor.Tensor.t
val relu : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val sat01 : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val (!.) : Base.float -> Ocannl_tensor.Tensor.t
val (!..) : ?label:Base.string Base.list -> int -> Ocannl_tensor.Tensor.t
val (!%) : ?label:Base.string Base.list -> Base.int64 -> Ocannl_tensor.Tensor.t
val (!@) : ?label:Base.string list -> Ir.Indexing.static_symbol -> Ocannl_tensor.Tensor.t
val dim : ?label:Base.string list -> Ocannl_tensor.Shape.delayed_var_ref -> Ocannl_tensor.Tensor.t
val (-) : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val (~-) : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val (/.) : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val exp : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val log : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val log2 : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val sin : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val cos : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val neg : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val sqrt : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val recip : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val recip_sqrt : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val tanh : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val not : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val (<) : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val (=) : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val (<>) : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val embed_self_id : ?label:Base.string list -> unit -> Ocannl_tensor.Tensor.t
val einsum : ?label:Base.string Base.list -> ?capture_dims:Ocannl_tensor.Shape.delayed_var_ref Base.list -> Base.string -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val outer_sum : ?label:Base.string Base.list -> ?capture_dims:Ocannl_tensor.Shape.delayed_var_ref Base.list -> Base.string -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val einsum1 : ?label:Base.string Base.list -> ?capture_dims:Ocannl_tensor.Shape.delayed_var_ref Base.list -> Base.string -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val einmax1 : ?label:Base.string Base.list -> ?capture_dims:Ocannl_tensor.Shape.delayed_var_ref Base.list -> Base.string -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val tropical : ?label:Base.string Base.list -> ?capture_dims:Ocannl_tensor.Shape.delayed_var_ref Base.list -> Base.string -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val offsets : ?label:Base.string Base.list -> unit -> Ocannl_tensor.Tensor.t
val uniform : ?label:Base.string Base.list -> unit -> Ocannl_tensor.Tensor.t
val uniform_at : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val uniform1 : ?label:Base.string Base.list -> unit -> Ocannl_tensor.Tensor.t
val uniform_at1 : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val normal : ?label:Base.string Base.list -> unit -> Ocannl_tensor.Tensor.t
val normal1 : ?label:Base.string Base.list -> unit -> Ocannl_tensor.Tensor.t
val normal_at : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t
val normal_at1 : ?label:Base.string Base.list -> Ocannl_tensor.Tensor.t -> Ocannl_tensor.Tensor.t