docs menu

Module Nx.Host

The process's host, Rig.host, computed by nx.cpu.

#type d = host

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.