DeviceKind

public enum DeviceKind
extension DeviceKind: Equatable

The kind of devices the tensor resides on.

  • CPU

    Declaration

    Swift

    case CPU
  • Declaration

    Swift

    case GPU(Int)
  • GPU device related information.

    See more

    Declaration

    Swift

    public enum GPUs
  • Declaration

    Swift

    public static func == (lhs: `Self`, rhs: `Self`) -> Bool