public struct Device
extension Device: Equatable
extension Device: CustomStringConvertible
Устройство, на котором можно разместить Tensor
.
Тип устройства: графический процессор, графический процессор, TPU или удаленный TPU.
Декларация
public let kind: Kind
Порядковый номер устройства.
Декларация
public let ordinal: Int
Бэкэнд, используемый для отправки тензорных операций.
Декларация
public let backend: Backend
Декларация
public init(kind: Kind, ordinal: Int, backend: Backend = defaultBackend)
Бэкэнд, используемый для диспетчеризации тензорных операций.
Декларация
public enum Backend
Тип устройства.
Декларация
public enum Kind
Декларация
public var isRemote: Bool { get }
Декларация
public static var defaultBackend: Backend { get }
Device
по умолчанию.Декларация
public static var `default`: Device { get }
Устройство XLA по умолчанию.
Декларация
public static var defaultXLA: Device { get }
Текущее устройство TF Eager.
Декларация
public static var defaultTFEager: Device { get }
Массив всех устройств.
Декларация
public static var allDevices: [Device] { get }
Декларация
public static func setReplicationDevices(_ devices: [Device])
Декларация
public static func getReplicationDevices() -> [Device]
Декларация
public static func syncLiveTensorsForDevices(_ devices: [Device])
Декларация
public static func == (lhs: Device, rhs: Device) -> Bool
Декларация
public var description: String { get }
Декларация
public var annotationsAvailable: String { get }
Список устройств, используемых для обучения.
Декларация
public static var trainingDevices: [Device] { get }
Список устройств, используемых для суммирования перекрестных реплик при обучении на TrainingDevices.
Декларация
public static var crossReplicaSumDevices: [Device] { get }