TensorFlow 2.0 Beta is available Learn more

Module: tf.tpu.experimental

Classes

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

Functions

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