Save the date! Google I/O returns May 18-20 Register now

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