Index

A

add_task() (ConcurrentEventLoop method) all_gather() (in module graphlearn_torch.distributed.rpc) amount (NegativeSampling attribute) append_cpu_tensor() (UnifiedTensor method)

append_shared_tensor() (UnifiedTensor method) apply_to_all_tensor() (in module graphlearn_torch.utils.tensor) assign_device() (in module graphlearn_torch.utils.device) async_get() (DistFeature method) async_request_server() (in module graphlearn_torch.distributed.dist_client)

B

barrier() (in module graphlearn_torch.distributed.rpc) BaseSampler (class in graphlearn_torch.sampler.base) batch (HeteroSamplerOutput attribute) (SamplerOutput attribute)

batch_size (SamplingConfig attribute) binary (NegativeSamplingMode attribute)

C

call() (RpcCalleeBase method) (RpcFeatureLookupCallee method) (RpcSamplingCallee method) (RpcSubGraphCallee method) cast() (CastMixin class method) CastMixin (class in graphlearn_torch.utils.mixin) ChannelBase (class in graphlearn_torch.channel.base) CLIENT (DistRole attribute) col (EdgeSamplerInput attribute) (HeteroSamplerOutput attribute) (SamplerOutput attribute)

col_count (Graph property) collect_features (SamplingConfig attribute) CollocatedDistSamplingWorkerOptions (class in graphlearn_torch.distributed.dist_options) ConcurrentEventLoop (class in graphlearn_torch.distributed.event_loop) convert_to_tensor() (in module graphlearn_torch.utils.tensor) cpu_get() (Feature method) create_inducer() (NeighborSampler method) create_sampling_producer() (DistServer method) CSRTopo (class in graphlearn_torch.data.graph)

D

Dataset (class in graphlearn_torch.data.dataset) degrees (CSRTopo property) destroy_sampling_producer() (DistServer method) device (HeteroSamplerOutput attribute) (SamplerOutput attribute) (UnifiedTensor property) DeviceGroup (class in graphlearn_torch.data.feature) DistCollocatedSamplingProducer (class in graphlearn_torch.distributed.dist_sampling_producer) DistContext (class in graphlearn_torch.distributed.dist_context)

DistDataset (class in graphlearn_torch.distributed.dist_dataset) DistFeature (class in graphlearn_torch.distributed.dist_feature) DistGraph (class in graphlearn_torch.distributed.dist_graph) DistMpSamplingProducer (class in graphlearn_torch.distributed.dist_sampling_producer) DistNeighborLoader (class in graphlearn_torch.distributed.dist_neighbor_loader) DistNeighborSampler (class in graphlearn_torch.distributed.dist_neighbor_sampler) DistRole (class in graphlearn_torch.distributed.dist_context) DistServer (class in graphlearn_torch.distributed.dist_server) drop_last (SamplingConfig attribute)

E

e_id (EdgeIndex attribute) edge (HeteroSamplerOutput attribute) (NeighborOutput attribute) (SamplerOutput attribute) edge_count (CSRTopo property) (Graph property) edge_feat_pb (DistDataset property)

edge_ids (CSRTopo property) edge_index (EdgeIndex attribute) edge_types (HeteroSamplerOutput attribute) EdgeIndex (class in graphlearn_torch.sampler.base) EdgeSamplerInput (class in graphlearn_torch.sampler.base) ensure_device() (in module graphlearn_torch.utils.device) ensure_dir() (in module graphlearn_torch.utils.common) exit() (DistServer method)

F

Feature (class in graphlearn_torch.data.feature) fetch_one_sampled_message() (DistServer method) format_hetero_sampler_output() (in module graphlearn_torch.utils.common) from_ipc_handle() (Dataset class method) (DistDataset class method) (Feature class method) (Graph class method) (UnifiedTensor method)

G

get_available_device() (in module graphlearn_torch.utils.device) get_context() (in module graphlearn_torch.distributed.dist_context) get_dataset_meta() (DistServer method) get_edge_feature() (Dataset method) get_edge_index() (HeteroSamplerOutput method) get_edge_partitions() (DistGraph method) get_edge_types() (Dataset method) get_free_port() (in module graphlearn_torch.utils.common) get_graph() (Dataset method) get_inducer() (NeighborSampler method) get_local_graph() (DistGraph method) get_node_feature() (Dataset method) get_node_label() (Dataset method) get_node_partitions() (DistGraph method) get_node_types() (Dataset method) get_rpc_current_group_worker_names() (in module graphlearn_torch.distributed.rpc) get_rpc_master_addr() (in module graphlearn_torch.distributed.rpc) get_rpc_master_port() (in module graphlearn_torch.distributed.rpc) get_rpc_worker_names() (in module graphlearn_torch.distributed.rpc) get_server() (in module graphlearn_torch.distributed.dist_server) get_to_worker() (RpcDataPartitionRouter method) global_all_gather() (in module graphlearn_torch.distributed.rpc) global_barrier() (in module graphlearn_torch.distributed.rpc) Graph (class in graphlearn_torch.data.graph) graph_handler (Graph property) graphlearn_torch.channel.base module graphlearn_torch.channel.mp_channel module graphlearn_torch.channel.remote_channel module graphlearn_torch.channel.shm_channel module graphlearn_torch.data.dataset module graphlearn_torch.data.feature module graphlearn_torch.data.graph module graphlearn_torch.data.reorder module graphlearn_torch.data.table_dataset module graphlearn_torch.data.unified_tensor module graphlearn_torch.data.vineyard_utils module

graphlearn_torch.distributed.dist_client module graphlearn_torch.distributed.dist_context module graphlearn_torch.distributed.dist_dataset module graphlearn_torch.distributed.dist_feature module graphlearn_torch.distributed.dist_graph module graphlearn_torch.distributed.dist_neighbor_loader module graphlearn_torch.distributed.dist_neighbor_sampler module graphlearn_torch.distributed.dist_options module graphlearn_torch.distributed.dist_sampling_producer module graphlearn_torch.distributed.dist_server module graphlearn_torch.distributed.event_loop module graphlearn_torch.distributed.rpc module graphlearn_torch.loader.neighbor_loader module graphlearn_torch.loader.transform module graphlearn_torch.sampler.base module graphlearn_torch.sampler.negative_sampler module graphlearn_torch.sampler.neighbor_sampler module graphlearn_torch.utils.common module graphlearn_torch.utils.device module graphlearn_torch.utils.exit_status module graphlearn_torch.utils.mixin module graphlearn_torch.utils.singleton module graphlearn_torch.utils.tensor module graphlearn_torch.utils.units module

H

HeteroSamplerOutput (class in graphlearn_torch.sampler.base)

I

id2idx() (in module graphlearn_torch.utils.tensor) id2idx_v2() (in module graphlearn_torch.utils.common) index (PartialNeighborOutput attribute) index_select() (in module graphlearn_torch.utils.common) indices (CSRTopo property) indptr (CSRTopo property) init() (DistCollocatedSamplingProducer method) (DistMpSamplingProducer method) init_client() (in module graphlearn_torch.distributed.dist_client) init_edge_features() (Dataset method) init_from() (UnifiedTensor method) init_graph() (Dataset method)

init_node_features() (Dataset method) init_node_labels() (Dataset method) init_rpc() (in module graphlearn_torch.distributed.rpc) init_server() (in module graphlearn_torch.distributed.dist_server) init_worker_group() (in module graphlearn_torch.distributed.dist_context) input_type (EdgeSamplerInput attribute) (HeteroSamplerOutput attribute) (NodeSamplerInput attribute) is_binary() (NegativeSampling method) is_client() (DistContext method) is_server() (DistContext method) is_triplet() (NegativeSampling method) is_worker() (DistContext method)

L

label (EdgeSamplerInput attribute) lazy_init() (Graph method) lazy_init_neg_sampler() (NeighborSampler method) lazy_init_sampler() (NeighborSampler method) lazy_init_subgraph_op() (NeighborSampler method)

lazy_init_with_ipc_handle() (Feature method) LINK (SamplingType attribute) load() (DistDataset method) load_edge_feature_from_vineyard() (in module graphlearn_torch.data.vineyard_utils) load_vertex_feature_from_vineyard() (in module graphlearn_torch.data.vineyard_utils) local_get() (DistFeature method)

M

merge_dict() (in module graphlearn_torch.utils.common) merge_hetero_sampler_output() (in module graphlearn_torch.utils.common) metadata (HeteroSamplerOutput attribute) (SamplerOutput attribute) mode (NegativeSampling attribute) module graphlearn_torch.channel.base graphlearn_torch.channel.mp_channel graphlearn_torch.channel.remote_channel graphlearn_torch.channel.shm_channel graphlearn_torch.data.dataset graphlearn_torch.data.feature graphlearn_torch.data.graph graphlearn_torch.data.reorder graphlearn_torch.data.table_dataset graphlearn_torch.data.unified_tensor graphlearn_torch.data.vineyard_utils graphlearn_torch.distributed.dist_client graphlearn_torch.distributed.dist_context graphlearn_torch.distributed.dist_dataset graphlearn_torch.distributed.dist_feature graphlearn_torch.distributed.dist_graph graphlearn_torch.distributed.dist_neighbor_loader graphlearn_torch.distributed.dist_neighbor_sampler graphlearn_torch.distributed.dist_options graphlearn_torch.distributed.dist_sampling_producer graphlearn_torch.distributed.dist_server graphlearn_torch.distributed.event_loop graphlearn_torch.distributed.rpc graphlearn_torch.loader.neighbor_loader graphlearn_torch.loader.transform graphlearn_torch.sampler.base graphlearn_torch.sampler.negative_sampler graphlearn_torch.sampler.neighbor_sampler graphlearn_torch.utils.common graphlearn_torch.utils.device graphlearn_torch.utils.exit_status graphlearn_torch.utils.mixin graphlearn_torch.utils.singleton graphlearn_torch.utils.tensor graphlearn_torch.utils.units

MP_STATUS_CHECK_INTERVAL (in module graphlearn_torch.distributed.dist_sampling_producer) MpChannel (class in graphlearn_torch.channel.mp_channel) MpCommand (class in graphlearn_torch.distributed.dist_sampling_producer) MpDistSamplingWorkerOptions (class in graphlearn_torch.distributed.dist_options)

N

nbr (NeighborOutput attribute) nbr_num (NeighborOutput attribute) neg_sampling (EdgeSamplerInput attribute) NegativeSampling (class in graphlearn_torch.sampler.base) NegativeSamplingMode (class in graphlearn_torch.sampler.base) NeighborLoader (class in graphlearn_torch.loader.neighbor_loader) NeighborOutput (class in graphlearn_torch.sampler.base) NeighborSampler (class in graphlearn_torch.sampler.neighbor_sampler) new_from_ipc() (UnifiedTensor class method)

node (HeteroSamplerOutput attribute) (NodeSamplerInput attribute) (SamplerOutput attribute) NODE (SamplingType attribute) node_feat_pb (DistDataset property) NodeSamplerInput (class in graphlearn_torch.sampler.base) num_clients() (DistContext method) num_neighbors (SamplingConfig attribute) num_servers() (DistContext method) numel (UnifiedTensor property)

O

output (PartialNeighborOutput attribute)

P

parse_size() (in module graphlearn_torch.utils.units) PartialNeighborOutput (class in graphlearn_torch.distributed.dist_neighbor_sampler)

pin_memory() (ShmChannel method) produce_all() (DistMpSamplingProducer method) python_exit_status (in module graphlearn_torch.utils.exit_status)

R

RANDOM_WALK (SamplingType attribute) RandomNegativeSampler (class in graphlearn_torch.sampler.negative_sampler) rebuild_dataset() (in module graphlearn_torch.data.dataset) rebuild_dist_dataset() (in module graphlearn_torch.distributed.dist_dataset) rebuild_feature() (in module graphlearn_torch.data.feature) rebuild_graph() (in module graphlearn_torch.data.graph) recv() (ChannelBase method) (MpChannel method) (RemoteReceivingChannel method) (ShmChannel method) reduce_dataset() (in module graphlearn_torch.data.dataset) reduce_dist_dataset() (in module graphlearn_torch.distributed.dist_dataset) reduce_feature() (in module graphlearn_torch.data.feature) reduce_graph() (in module graphlearn_torch.data.graph) RemoteDistSamplingWorkerOptions (class in graphlearn_torch.distributed.dist_options) RemoteReceivingChannel (class in graphlearn_torch.channel.remote_channel) request_server() (in module graphlearn_torch.distributed.dist_client) reset() (DistCollocatedSamplingProducer method) (RemoteReceivingChannel method)

row (EdgeSamplerInput attribute) (HeteroSamplerOutput attribute) (SamplerOutput attribute) row_count (CSRTopo property) (Graph property) rpc_global_request() (in module graphlearn_torch.distributed.rpc) rpc_global_request_async() (in module graphlearn_torch.distributed.rpc) rpc_is_initialized() (in module graphlearn_torch.distributed.rpc) rpc_register() (in module graphlearn_torch.distributed.rpc) rpc_request() (in module graphlearn_torch.distributed.rpc) rpc_request_async() (in module graphlearn_torch.distributed.rpc) rpc_sync_data_partitions() (in module graphlearn_torch.distributed.rpc) RpcCalleeBase (class in graphlearn_torch.distributed.rpc) RpcDataPartitionRouter (class in graphlearn_torch.distributed.rpc) RpcFeatureLookupCallee (class in graphlearn_torch.distributed.dist_feature) RpcSamplingCallee (class in graphlearn_torch.distributed.dist_neighbor_sampler) RpcSubGraphCallee (class in graphlearn_torch.distributed.dist_neighbor_sampler) run_task() (ConcurrentEventLoop method)

S

sample() (DistCollocatedSamplingProducer method) (RandomNegativeSampler method) SAMPLE_ALL (MpCommand attribute) sample_from_edges() (BaseSampler method) (DistNeighborSampler method) (NeighborSampler method) sample_from_nodes() (BaseSampler method) (DistNeighborSampler method) (NeighborSampler method) sample_one_hop() (NeighborSampler method) sample_prob() (NeighborSampler method) sample_pyg_v1() (NeighborSampler method) SamplerOutput (class in graphlearn_torch.sampler.base) sampling_type (SamplingConfig attribute) SamplingConfig (class in graphlearn_torch.sampler.base) SamplingType (class in graphlearn_torch.sampler.base) send() (ChannelBase method) (MpChannel method) (RemoteReceivingChannel method) (ShmChannel method) SERVER (DistRole attribute) SERVER_EXIT_STATUS_CHECK_INTERVAL (in module graphlearn_torch.distributed.dist_server) shape (Feature property) (UnifiedTensor property) share_ipc() (Dataset method) (DistDataset method) (Feature method) (Graph method) (UnifiedTensor method)

share_memory() (EdgeSamplerInput method) (in module graphlearn_torch.utils.tensor) (NegativeSampling method) (NodeSamplerInput method) share_memory_() (CSRTopo method) ShmChannel (class in graphlearn_torch.channel.shm_channel) shuffle (SamplingConfig attribute) shutdown() (DistCollocatedSamplingProducer method) (DistMpSamplingProducer method) (DistServer method) shutdown_client() (in module graphlearn_torch.distributed.dist_client) shutdown_loop() (ConcurrentEventLoop method) shutdown_rpc() (in module graphlearn_torch.distributed.rpc) singleton() (in module graphlearn_torch.utils.singleton) size (DeviceGroup property) (EdgeIndex attribute) size() (Feature method) (UnifiedTensor method) sort_by_in_degree() (in module graphlearn_torch.data.reorder) squeeze() (in module graphlearn_torch.utils.tensor) start_loop() (ConcurrentEventLoop method) start_new_epoch_sampling() (DistServer method) STOP (MpCommand attribute) stride() (UnifiedTensor method) SUBGRAPH (SamplingType attribute) subgraph() (BaseSampler method) (DistNeighborSampler method) (NeighborSampler method) subgraph_op (NeighborSampler property)

T

TableDataset (class in graphlearn_torch.data.table_dataset) tensor_equal_with_device() (in module graphlearn_torch.utils.tensor) to() (EdgeIndex method) (EdgeSamplerInput method) (NegativeSampling method) (NeighborOutput method) (NodeSamplerInput method)

to_coo() (CSRTopo method) to_csc() (CSRTopo method) to_data() (in module graphlearn_torch.loader.transform) to_hetero_data() (in module graphlearn_torch.loader.transform) triplet (NegativeSamplingMode attribute)

U

UnifiedTensor (class in graphlearn_torch.data.unified_tensor)

V

vineyard_to_csr() (in module graphlearn_torch.data.vineyard_utils)

W

wait_all() (ConcurrentEventLoop method) wait_and_shutdown_server() (in module graphlearn_torch.distributed.dist_server) wait_for_exit() (DistServer method) weight (NegativeSampling attribute)

with_edge (SamplingConfig attribute) with_neg (SamplingConfig attribute) WORKER (DistRole attribute) worker_name (DistContext property) wrap_torch_future() (in module graphlearn_torch.distributed.event_loop)