Устройство

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 }