docs menu

Module type Nx.Devices

The type for device sets as modules: a brand and its placements.

#type d

The set's brand.

#val v : d Devices.t

v is the set.

#val on : d Placement.t

on is Placement.on v.

#val split : axis:int -> d Placement.t

split ~axis is Placement.split ~axis v.