Up – Package index » torch » Sources » torch.core » device.ml

Source file `device.ml`

torch
- README
- Library torch
  - Torch
- Library torch.core
  - Torch_core
- Library torch.toplevel
  - Torch_toplevel
- Library torch.vision
  - Torch_vision
- Sources

1
2
3
4
5
6
type t =
  | Cpu
  | Cuda

(* Hardcoded, should match Device.h *)
let to_int = function Cpu -> 0 | Cuda -> 1