package torch

You can search for identifiers within the package.

in-package search v0.2.0

torch
- Library torch
  - Torch
    
    Checkpointing
    
    Cifar_helper
    
    Cuda
    
    Dataset_helper
    
    Device
    
    Ivalue
    
    Layer
    
    Lstm
    
    Mnist_helper
    
    Module
    
    Optimizer
    
    Clip_grad
    
    Linear_interpolation
    
    Scalar
    
    Serialize
    
    Tensor
    
    Text_helper
    
    Var_store
    
    Init
- Library torch.core
  - Torch_core
    
    Device
    
    Kind
    
    Reduction
    
    Torch_bindings
    
    C
    
    Tensor
    
    Scalar
    
    Serialize
    
    Optimizer
    
    Cuda
    
    Ivalue
    
    Module
    
    TensorG
    
    Torch_bindings_generated
    
    C
    
    Torch_generated
    
    CI
    
    Wrapper
    
    Scalar
    
    Tensor
    
    Optimizer
    
    Serialize
    
    Cuda
    
    Ivalue
    
    Tag
    
    Module
    
    Wrapper_generated
    
    C
    
    Tensor
    
    Scalar
    
    Serialize
    
    Optimizer
    
    Cuda
    
    Ivalue
    
    Module
    
    TensorG
    
    Wrapper_generated_intf
    
    S
- Library torch.toplevel
  - Torch_toplevel
- Library torch.vision
  - Torch_vision
    
    Alexnet
    
    Densenet
    
    Image
    
    Loader
    
    Imagenet
    
    Classes
    
    Loader
    
    Inception
    
    Mobilenet
    
    Resnet
    
    Squeezenet
    
    Stb_image_write
    
    Vgg
- Sources
  - torch
    
    checkpointing.ml
    
    cifar_helper.ml
    
    cuda.ml
    
    dataset_helper.ml
    
    device.ml
    
    ivalue.ml
    
    layer.ml
    
    mnist_helper.ml
    
    module.ml
    
    optimizer.ml
    
    scalar.ml
    
    serialize.ml
    
    tensor.ml
    
    text_helper.ml
    
    torch.ml
    
    var_store.ml
  - torch.core
    
    device.ml
    
    kind.ml
    
    reduction.ml
    
    torch_bindings.ml
    
    torch_bindings_generated.ml
    
    torch_core.ml
    
    torch_generated.ml
    
    wrapper.ml
    
    wrapper_generated.ml
    
    wrapper_generated_intf.ml
  - torch.toplevel
    
    torch_toplevel.ml
  - torch.vision
    
    alexnet.ml
    
    densenet.ml
    
    image.ml
    
    imagenet.ml
    
    inception.ml
    
    mobilenet.ml
    
    resnet.ml
    
    squeezenet.ml
    
    stb_image_write.ml
    
    torch_vision.ml
    
    vgg.ml

Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Source file `device.ml`

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

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

package torch

Source file device.ml

Source file `device.ml`