File data_parallel.h

↰ Parent directory (torch/csrc/api/include/torch/nn/parallel)

Definition (torch/csrc/api/include/torch/nn/parallel/data_parallel.h)

• Program Listing for File data_parallel.h

Includes

• ATen/Device.h

• ATen/Parallel.h

• ATen/core/functional.h

• c10/core/TensorOptions.h

• c10/util/Exception.h (File Exception.h)

• c10/util/irange.h

• exception (File worker_exception.h)

• memory

• mutex

• torch/csrc/autograd/functions/comm.h

• torch/csrc/autograd/functions/utils.h

• torch/cuda.h (File cuda.h)

• torch/nn/module.h (File module.h)

• torch/nn/pimpl.h (File pimpl.h)

• torch/types.h (File types.h)

• vector

Namespaces

• Namespace torch

• Namespace torch::nn

• Namespace torch::nn::parallel

Functions

• Template Function torch::nn::parallel::data_parallel

• Template Function torch::nn::parallel::parallel_apply

• Template Function torch::nn::parallel::replicate(const std::shared_ptr<ModuleType>&, const std::vector<Device>&)

• Template Function torch::nn::parallel::replicate(const ModuleHolder<ModuleType>&, const std::vector<Device>&)

• Template Function torch::nn::replicate_grad_edges

Variables

• Variable destination_device_