DeviceKind

public enum DeviceKind

A TensorFlow device kind.

  • cpu

    The CPU device kind.

    Declaration

    case cpu
  • gpu

    The GPU device kind.

    Declaration

    case gpu
  • tpu

    The TPU device kind.

    Declaration

    case tpu