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 phụ trợ đượ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 }