(x, n_devices)
return tl.reshape_by_device(x, n_devices)
Reshapes possibly nested x into a shape (n_devices, ...).
Reshapes possibly nested x into a shape (n_devices, ...).
Reshapes possibly nested x into a shape ( n_devices ... ) .
def _reshape_by_device(x, n_devices):
  """Reshapes possibly nested x into a shape (n_devices, ...)."""
  return tl.reshape_by_device(x, n_devices)
def _reshape_by_device ( x , n_devices ) : return tl . reshape_by_device ( x , n_devices )