Public instance methods
string cpu_device_name_at_coordinates(ValueTuple device_coordinates, object job)
object cpu_device_name_at_coordinates_dyn(object device_coordinates, object job)
object task_ordinal_at_coordinates_dyn(object device_coordinates)
string tpu_device_name_at_coordinates(ValueTuple device_coordinates, object job)
object tpu_device_name_at_coordinates_dyn(object device_coordinates, object job)
object tpu_device_ordinal_at_coordinates(ValueTuple device_coordinates)
object tpu_device_ordinal_at_coordinates_dyn(object device_coordinates)
Public properties
object device_coordinates_dyn get;
object mesh_rank_dyn get;
object mesh_shape_dyn get;
object num_tasks_dyn get;
object num_tpus_per_task get;
object num_tpus_per_task_dyn get;