Index

A

add_step_closure() (in module torch_xla.core.xla_model) all_gather() (in module torch_xla.core.functions) (in module torch_xla.core.xla_model)

all_reduce() (in module torch_xla.core.functions) (in module torch_xla.core.xla_model) all_to_all() (in module torch_xla.core.xla_model)

D

DataWrapper (class in torch_xla.utils.utils)

do_on_ordinals() (in module torch_xla.core.xla_model)

G

get_local_ordinal() (in module torch_xla.core.xla_model) get_memory_info() (in module torch_xla.core.xla_model) get_ordinal() (in module torch_xla.core.xla_model)

get_rng_state() (in module torch_xla.core.xla_model) get_stablehlo() (in module torch_xla.core.xla_model) get_stablehlo_bytecode() (in module torch_xla.core.xla_model) get_xla_supported_devices() (in module torch_xla.core.xla_model)

I

is_master_ordinal() (in module torch_xla.core.xla_model)

L

load() (in module torch_xla.utils.serialization)

M

mesh_reduce() (in module torch_xla.core.xla_model) module torch_xla.core.functions torch_xla.core.xla_model torch_xla.distributed.parallel_loader torch_xla.distributed.xla_multiprocessing torch_xla.utils.serialization torch_xla.utils.utils

MpModelWrapper (class in torch_xla.distributed.xla_multiprocessing) MpSerialExecutor (class in torch_xla.distributed.xla_multiprocessing)

O

optimizer_step() (in module torch_xla.core.xla_model)

P

ParallelLoader (class in torch_xla.distributed.parallel_loader)

per_device_loader() (torch_xla.distributed.parallel_loader.ParallelLoader method)

R

rendezvous() (in module torch_xla.core.xla_model)

run() (torch_xla.distributed.xla_multiprocessing.MpSerialExecutor method)

S

SampleGenerator (class in torch_xla.utils.utils) save() (in module torch_xla.core.xla_model) (in module torch_xla.utils.serialization)

set_rng_state() (in module torch_xla.core.xla_model) spawn() (in module torch_xla.distributed.xla_multiprocessing)

T

to() (torch_xla.distributed.xla_multiprocessing.MpModelWrapper method) torch_xla.core.functions module torch_xla.core.xla_model module torch_xla.distributed.parallel_loader module

torch_xla.distributed.xla_multiprocessing module torch_xla.utils.serialization module torch_xla.utils.utils module

W

wait_device_ops() (in module torch_xla.core.xla_model)

X

xla_device() (in module torch_xla.core.xla_model)

xla_device_hw() (in module torch_xla.core.xla_model) xrt_world_size() (in module torch_xla.core.xla_model)