public struct Device
extension Device: Equatable
extension Device: CustomStringConvertible
Um dispositivo no qual Tensor
s pode ser alocado.
O tipo de dispositivo: GPU, GPU, TPU ou TPU remoto.
Declaração
public let kind: Kind
O valor ordinal do dispositivo.
Declaração
public let ordinal: Int
O back-end usado para despachar as operações do tensor.
Declaração
public let backend: Backend
Declaração
public init(kind: Kind, ordinal: Int, backend: Backend = defaultBackend)
Back-end usado para despachar as operações do tensor.
Declaração
public enum Backend
Um tipo de dispositivo.
Declaração
public enum Kind
Declaração
public var isRemote: Bool { get }
Declaração
public static var defaultBackend: Backend { get }
O
Device
padrão.Declaração
public static var `default`: Device { get }
O dispositivo XLA padrão.
Declaração
public static var defaultXLA: Device { get }
O atual dispositivo TF Eager.
Declaração
public static var defaultTFEager: Device { get }
Uma matriz de todos os dispositivos.
Declaração
public static var allDevices: [Device] { get }
Declaração
public static func setReplicationDevices(_ devices: [Device])
Declaração
public static func getReplicationDevices() -> [Device]
Declaração
public static func syncLiveTensorsForDevices(_ devices: [Device])
Declaração
public static func == (lhs: Device, rhs: Device) -> Bool
Declaração
public var description: String { get }
Declaração
public var annotationsAvailable: String { get }
Uma lista de dispositivos usados para treinamento.
Declaração
public static var trainingDevices: [Device] { get }
Uma lista de dispositivos usados para somas de réplicas cruzadas ao treinar em trainingDevices.
Declaração
public static var crossReplicaSumDevices: [Device] { get }