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 }