Missed TensorFlow World? Check out the recap. Learn more

Module: tf.tpu.experimental

TensorFlow 2.0 version

Classes

class DeviceAssignment: Mapping from logical cores in a computation to the physical TPU topology.

Functions

initialize_tpu_system(...): Initialize the TPU devices.