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)

C

CachedDataset (class in torch_xla.utils.cached_dataset)

D

DataWrapper (class in torch_xla.utils.utils)

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

G

generic_open() (in module torch_xla.utils.gcsfs) generic_read() (in module torch_xla.utils.gcsfs) generic_write() (in module torch_xla.utils.gcsfs) 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_xla_supported_devices() (in module torch_xla.core.xla_model)

I

is_gcs_path() (in module torch_xla.utils.gcsfs)

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

L

list() (in module torch_xla.utils.gcsfs)

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

M

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

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

N

nms() (in module torch_xla.core.functions)

O

open() (in module torch_xla.utils.gcsfs)

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

read() (in module torch_xla.utils.gcsfs) remove() (in module torch_xla.utils.gcsfs)

rendezvous() (in module torch_xla.core.xla_model) rmtree() (in module torch_xla.utils.gcsfs) 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) stat() (in module torch_xla.utils.gcsfs)

T

TfRecordReader (class in torch_xla.utils.tf_record_reader) 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.cached_dataset (module) torch_xla.utils.gcsfs (module) torch_xla.utils.serialization (module) torch_xla.utils.tf_record_reader (module) torch_xla.utils.utils (module)

W

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

write() (in module torch_xla.utils.gcsfs)

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)