Thiết bị

public struct Device
extension Device: Equatable
extension Device: CustomStringConvertible

Một thiết bị mà Tensor s có thể được phân bổ trên đó.

  • Loại thiết bị: GPU, GPU, TPU hoặc TPU từ xa.

    Tuyên ngôn

    public let kind: Kind
  • Giá trị thứ tự của thiết bị.

    Tuyên ngôn

    public let ordinal: Int
  • Phần phụ trợ được sử dụng để gửi các hoạt động tensor.

    Tuyên ngôn

    public let backend: Backend
  • Tuyên ngôn

    public init(kind: Kind, ordinal: Int, backend: Backend = defaultBackend)
  • Phần cuối được sử dụng để gửi các hoạt động tensor.

    Tuyên ngôn

    public enum Backend
  • Một loại thiết bị.

    Tuyên ngôn

    public enum Kind
  • Tuyên ngôn

    public var isRemote: Bool { get }
  • Tuyên ngôn

    public static var defaultBackend: Backend { get }
  • Device mặc định.

    Tuyên ngôn

    public static var `default`: Device { get }
  • Thiết bị XLA mặc định.

    Tuyên ngôn

    public static var defaultXLA: Device { get }
  • Thiết bị TF Eager hiện tại.

    Tuyên ngôn

    public static var defaultTFEager: Device { get }
  • Một loạt các thiết bị.

    Tuyên ngôn

    public static var allDevices: [Device] { get }
  • Tuyên ngôn

    public static func setReplicationDevices(_ devices: [Device])
  • Tuyên ngôn

    public static func getReplicationDevices() -> [Device]
  • Tuyên ngôn

    public static func syncLiveTensorsForDevices(_ devices: [Device])
  • Tuyên ngôn

    public static func == (lhs: Device, rhs: Device) -> Bool
  • Tuyên ngôn

    public var description: String { get }
  • Tuyên ngôn

    public var annotationsAvailable: String { get }
  • Danh sách các thiết bị được sử dụng để đào tạo.

    Tuyên ngôn

    public static var trainingDevices: [Device] { get }
  • Danh sách các thiết bị được sử dụng để tính tổng bản sao chéo khi đào tạo về Thiết bị đào tạo.

    Tuyên ngôn

    public static var crossReplicaSumDevices: [Device] { get }