parameters: (self)
return_statement: return self._device_coordinates.shape[1]
docstring: Returns the number of TPU devices per task in the TPU slice.
docstring_summary: Returns the number of TPU devices per task in the TPU slice.
docstring_tokens: Returns the number of TPU devices per task in the TPU slice .
function: def num_tpus_per_task(self):
      """Returns the number of TPU devices per task in the TPU slice."""
      return self._device_coordinates.shape[1]
function_tokens: def num_tpus_per_task ( self ) : return self . _device_coordinates . shape [ 1 ]