Dispositivo

public struct Device
extension Device: Equatable
extension Device: CustomStringConvertible

Um dispositivo no qual Tensor s podem ser alocados.

  • 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 de 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 de 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 padrão Device .

    Declaração

    public static var `default`: Device { get }
  • O dispositivo XLA padrão.

    Declaração

    public static var defaultXLA: Device { get }
  • O dispositivo TF Eager atual.

    Declaração

    public static var defaultTFEager: Device { get }
  • Uma variedade 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 durante o treinamento em dispositivos de treinamento.

    Declaração

    public static var crossReplicaSumDevices: [Device] { get }