1syntax = "proto3"; 2 3option cc_enable_arenas = true; 4 5package tensorflow.tpu; 6 7// Describes the geometry of a TPU mesh. 8message TopologyProto { 9 // The dimensions of the TPU topology, in cores. Typically, this is a 3D 10 // topology [x, y, core], where the major dimensions correspond to TPU chips, 11 // and the minor dimension describes the number of cores on a multicore chip. 12 repeated int32 mesh_shape = 1; 13 14 // Number of TensorFlow tasks in the cluster. 15 int32 num_tasks = 2; 16 17 // Number of TPU devices per task. 18 int32 num_tpu_devices_per_task = 3; 19 20 // A flattened rank 3 int32 array with shape 21 // [num_tasks, num_tpu_devices_per_task, len(mesh_shape)]. 22 // `tasks` is the number of tasks in the TPU cluster, `devices` is the number 23 // of TPU devices per task, and the minor dimension corresponds to a position 24 // in the TPU mesh topology. Each entry [task, device, axis] gives the 25 // `axis`-th coordinate in the topology of a task/device pair. 26 repeated int32 device_coordinates = 4; 27} 28