PyTorch
|
#include <c10/core/DeviceType.h>
#include <c10/macros/Macros.h>
#include <c10/util/Exception.h>
#include <cstddef>
#include <functional>
#include <iosfwd>
#include <string>
Go to the source code of this file.
Classes | |
struct | c10::Device |
Represents a a compute device on which a tensor is located. More... | |
Namespaces | |
namespace | c10 |
namespace | std |
Typedefs | |
using | c10::DeviceIndex = int8_t |
An index representing a specific device; e.g., the 1 in GPU 1. More... | |
Functions | |
std::ostream & | c10::operator<< (std::ostream &stream, const Device &device) |