Module PDSL.O

val (*) : ?label:Base.string Base.list -> Tensor.t -> Tensor.t -> Tensor.t
val (*.) : ?label:Base.string Base.list -> Tensor.t -> Tensor.t -> Tensor.t
val (+) : ?label:Base.string Base.list -> Tensor.t -> Tensor.t -> Tensor.t
val threefry4x32 : ?label:Base.string Base.list -> Tensor.t -> Tensor.t -> Tensor.t
val uint4x32_to_prec_uniform : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val uint4x32_to_prec_uniform1 : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val (**.) : ?label:Base.string Base.list -> Tensor.t -> Base.float -> Tensor.t
val relu : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val sat01 : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val fma : ?label:Base.string Base.list -> Tensor.t -> Tensor.t -> Tensor.t -> Tensor.t
val (!.) : Base.float -> Tensor.t
val (!..) : ?label:Base.string Base.list -> int -> Tensor.t
val (!%) : ?label:Base.string Base.list -> Base.int64 -> Tensor.t
val (!@) : ?label:Base.string list -> Ir.Indexing.static_symbol -> Tensor.t
val dim : ?label:Base.string list -> Shape.delayed_var_ref -> Tensor.t
val (-) : ?label:Base.string Base.list -> Tensor.t -> Tensor.t -> Tensor.t
val (~-) : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val (/.) : ?label:Base.string Base.list -> Tensor.t -> Tensor.t -> Tensor.t
val (@|) : ?label:Base.string Base.list -> Tensor.t -> Idx.static_symbol -> Tensor.t
val exp : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val log : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val log2 : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val sin : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val cos : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val neg : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val sqrt : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val recip : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val recip_sqrt : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val tanh : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val where : ?label:Base.string Base.list -> Tensor.t -> Tensor.t -> Tensor.t -> Tensor.t
val not : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val (<) : ?label:Base.string Base.list -> Tensor.t -> Tensor.t -> Tensor.t
val (=) : ?label:Base.string Base.list -> Tensor.t -> Tensor.t -> Tensor.t
val (<>) : ?label:Base.string Base.list -> Tensor.t -> Tensor.t -> Tensor.t
val embed_self_id : ?label:Base.string list -> unit -> Tensor.t
val einsum : ?label:Base.string Base.list -> ?capture_dims:Shape.delayed_var_ref Base.list -> Base.string -> Tensor.t -> Tensor.t -> Tensor.t
val outer_sum : ?label:Base.string Base.list -> ?capture_dims:Shape.delayed_var_ref Base.list -> Base.string -> Tensor.t -> Tensor.t -> Tensor.t
val einsum1 : ?label:Base.string Base.list -> ?capture_dims:Shape.delayed_var_ref Base.list -> Base.string -> Tensor.t -> Tensor.t
val einmax1 : ?label:Base.string Base.list -> ?capture_dims:Shape.delayed_var_ref Base.list -> Base.string -> Tensor.t -> Tensor.t
val tropical : ?label:Base.string Base.list -> ?capture_dims:Shape.delayed_var_ref Base.list -> Base.string -> Tensor.t -> Tensor.t -> Tensor.t
val offsets : ?label:Base.string Base.list -> unit -> Tensor.t
val uniform : ?label:Base.string Base.list -> unit -> Tensor.t
val uniform_at : ?label:Base.string Base.list -> Tensor.t -> Tensor.t
val uniform1 : ?label:Base.string Base.list -> unit -> Tensor.t
val uniform_at1 : ?label:Base.string Base.list -> Tensor.t -> Tensor.t