This Day in TensorFlow 2: tf.tpu

Image copyright TensorFlow

What is TensorFlow?

What is TPU?

Image copyright Google Cloud. Image Source

1. tf.tpu.experimental.initialize_tpu_system

tf.tpu.experimental.initialize_tpu_system(cluster_resolver=None)
  • cluster_resolver: provided information of the TPU cluster
  • There is a TPU: tf.tpu.Topology object for the topology(arrangement) of the TPU cluster
  • No TPU found: RuntimeError

2. tf.tpu.experimental.shutdown_tpu_system

tf.tpu.experimental.shutdown_tpu_system(cluster_resolver=None)
  • cluster_resolver: provided information of the TPU cluster
  • There is a TPU: tf.tpu.Topology object for the topology(arrangement) of the TPU cluster
  • No TPU found: RuntimeError if no devices found thru eager execution

- Aaron Ma, https://aaronhma.com

--

--

--

Living in The Milky Way 😀

Love podcasts or audiobooks? Learn on the go with our new app.

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store
Aaron Ma

Aaron Ma

Living in The Milky Way 😀

More from Medium

dentifying and marking image regions

User Authentication in Flask with JWT

Tensorflow series

Tumor Detection using CNN