Ppx_helper (p.arrayjit.0.6.0.doc.arrayjit.ppx_arrayjit.Ppx_arrayjit.Ppx

Sourcetype li = Ppxlib.longident

val collect_list : 
  Ppxlib_ast.Ast.expression Base.List.t ->
  Ppxlib_ast.Ast.expression ->
  Ppxlib_ast.Ast.expression Base.List.t

Source

val dim_spec_to_string : 
  [< `Batch_dims of Base.Int.t
  | `Input_dims of Base.Int.t
  | `Output_dims of Base.Int.t ] ->
  Base.String.t

Source

val ndarray_constant : 
  Ppxlib.expression ->
  Ppxlib.expression
  * Astlib.Ast_502.Parsetree.expression Base.List.t
  * Astlib.Ast_502.Parsetree.expression Base.List.t
  * Astlib.Ast_502.Parsetree.expression Base.List.t

Source

val substitute_identifiers_in_einsum_spec : 
  loc:Ppxlib.Location.t ->
  Base.String.t ->
  Astlib.Ast_502.Parsetree.expression

Convert a string containing patterns like "identifier*" to an OCaml expression that substitutes the identifiers with their runtime values. Identifiers match the pattern a-z_a-z0-9_* and must directly precede '*'.

Example usage: substitute_identifiers_in_string ~loc "a *x + b * y" generates an expression equivalent to: String.concat "" [Int.to_string a; " *x + "; Int.to_string b; " * y"]

So if a = 2 and b = 3, the result would be "2 *x + 3 * y". Whitespace between identifiers and '*' is preserved.