arrayjit 0.4.0 · OCaml Package

val physical_devices_only : [> `Physical_devices_only ] Variantslib.Variant.t

val for_parallel_copying : [> `For_parallel_copying ] Variantslib.Variant.t

val most_parallel_devices : [> `Most_parallel_devices ] Variantslib.Variant.t

val fold : 
  init:'a ->
  physical_devices_only:
    ('b -> [> `Physical_devices_only ] Variantslib.Variant.t -> 'c) ->
  for_parallel_copying:
    ('d -> [> `For_parallel_copying ] Variantslib.Variant.t -> 'e) ->
  most_parallel_devices:
    ('f -> [> `Most_parallel_devices ] Variantslib.Variant.t -> 'g) ->
  'h

val iter : 
  physical_devices_only:
    ([> `Physical_devices_only ] Variantslib.Variant.t -> unit) ->
  for_parallel_copying:
    ([> `For_parallel_copying ] Variantslib.Variant.t -> unit) ->
  most_parallel_devices:
    ([> `Most_parallel_devices ] Variantslib.Variant.t -> unit) ->
  unit

val map : 
  [< `For_parallel_copying | `Most_parallel_devices | `Physical_devices_only ] ->
  physical_devices_only:
    ([> `Physical_devices_only ] Variantslib.Variant.t -> 'a) ->
  for_parallel_copying:([> `For_parallel_copying ] Variantslib.Variant.t -> 'b) ->
  most_parallel_devices:
    ([> `Most_parallel_devices ] Variantslib.Variant.t -> 'c) ->
  'd

val make_matcher : 
  physical_devices_only:
    ([> `Physical_devices_only ] Variantslib.Variant.t ->
      'a ->
      (unit -> 'b) * 'c) ->
  for_parallel_copying:
    ([> `For_parallel_copying ] Variantslib.Variant.t ->
      'd ->
      (unit -> 'e) * 'f) ->
  most_parallel_devices:
    ([> `Most_parallel_devices ] Variantslib.Variant.t ->
      'g ->
      (unit -> 'h) * 'i) ->
  'j ->
  ([< `For_parallel_copying | `Most_parallel_devices | `Physical_devices_only ] ->
  'k)
  * 'l

val to_rank : 
  [< `For_parallel_copying | `Most_parallel_devices | `Physical_devices_only ] ->
  int

val to_name : 
  [< `For_parallel_copying | `Most_parallel_devices | `Physical_devices_only ] ->
  string

val descriptions : (string * int) list

package arrayjit