public struct Device
extension Device: Equatable
extension Device: CustomStringConvertible
Perangkat tempat Tensor
dapat dialokasikan.
Jenis perangkat: GPU, GPU, TPU, atau TPU jarak jauh.
Pernyataan
public let kind: Kind
Nilai ordinal perangkat.
Pernyataan
public let ordinal: Int
Backend digunakan untuk mengirimkan operasi tensor.
Pernyataan
public let backend: Backend
Pernyataan
public init(kind: Kind, ordinal: Int, backend: Backend = defaultBackend)
Backend digunakan untuk mengirimkan operasi tensor.
Pernyataan
public enum Backend
Jenis perangkat.
Pernyataan
public enum Kind
Pernyataan
public var isRemote: Bool { get }
Pernyataan
public static var defaultBackend: Backend { get }
Device
bawaan.Pernyataan
public static var `default`: Device { get }
Perangkat XLA bawaan.
Pernyataan
public static var defaultXLA: Device { get }
Perangkat TF Eager saat ini.
Pernyataan
public static var defaultTFEager: Device { get }
Serangkaian semua perangkat.
Pernyataan
public static var allDevices: [Device] { get }
Pernyataan
public static func setReplicationDevices(_ devices: [Device])
Pernyataan
public static func getReplicationDevices() -> [Device]
Pernyataan
public static func syncLiveTensorsForDevices(_ devices: [Device])
Pernyataan
public static func == (lhs: Device, rhs: Device) -> Bool
Pernyataan
public var description: String { get }
Pernyataan
public var annotationsAvailable: String { get }
Daftar perangkat yang digunakan untuk pelatihan.
Pernyataan
public static var trainingDevices: [Device] { get }
Daftar perangkat yang digunakan untuk penjumlahan replika silang saat pelatihan pada perangkat pelatihan.
Pernyataan
public static var crossReplicaSumDevices: [Device] { get }