• graphlearn_torch.channel
  • base
    • ChannelBase
      • ChannelBase.send()
      • ChannelBase.recv()
  • mp_channel
    • MpChannel
      • MpChannel.send()
      • MpChannel.recv()
  • remote_channel
    • RemoteReceivingChannel
      • RemoteReceivingChannel.reset()
      • RemoteReceivingChannel.send()
      • RemoteReceivingChannel.recv()
  • shm_channel
    • ShmChannel
      • ShmChannel.pin_memory()
      • ShmChannel.send()
      • ShmChannel.recv()
• graphlearn_torch.data
  • dataset
    • Dataset
      • Dataset.init_graph()
      • Dataset.init_node_features()
      • Dataset.init_edge_features()
      • Dataset.init_node_labels()
      • Dataset.share_ipc()
      • Dataset.from_ipc_handle()
      • Dataset.get_graph()
      • Dataset.get_node_types()
      • Dataset.get_edge_types()
      • Dataset.get_node_feature()
      • Dataset.get_edge_feature()
      • Dataset.get_node_label()
    • rebuild_dataset()
    • reduce_dataset()
  • feature
    • DeviceGroup
      • DeviceGroup.size
    • Feature
      • Feature.cpu_get()
      • Feature.share_ipc()
      • Feature.from_ipc_handle()
      • Feature.lazy_init_with_ipc_handle()
      • Feature.shape
      • Feature.size()
    • rebuild_feature()
    • reduce_feature()
  • graph
    • CSRTopo
      • CSRTopo.to_coo()
      • CSRTopo.to_csc()
      • CSRTopo.indptr
      • CSRTopo.indices
      • CSRTopo.edge_ids
      • CSRTopo.degrees
      • CSRTopo.row_count
      • CSRTopo.edge_count
      • CSRTopo.share_memory_()
    • Graph
      • Graph.lazy_init()
      • Graph.share_ipc()
      • Graph.from_ipc_handle()
      • Graph.row_count
      • Graph.col_count
      • Graph.edge_count
      • Graph.graph_handler
    • rebuild_graph()
    • reduce_graph()
  • reorder
    • sort_by_in_degree()
  • table_dataset
    • TableDataset
  • unified_tensor
    • UnifiedTensor
      • UnifiedTensor.append_shared_tensor()
      • UnifiedTensor.append_cpu_tensor()
      • UnifiedTensor.init_from()
      • UnifiedTensor.shape
      • UnifiedTensor.device
      • UnifiedTensor.numel
      • UnifiedTensor.size()
      • UnifiedTensor.stride()
      • UnifiedTensor.share_ipc()
      • UnifiedTensor.from_ipc_handle()
      • UnifiedTensor.new_from_ipc()
  • vineyard_utils
    • vineyard_to_csr()
    • load_vertex_feature_from_vineyard()
    • load_edge_feature_from_vineyard()
• graphlearn_torch.distributed
  • dist_client
    • init_client()
    • shutdown_client()
    • async_request_server()
    • request_server()
  • dist_context
    • DistRole
      • DistRole.WORKER
      • DistRole.SERVER
      • DistRole.CLIENT
    • DistContext
      • DistContext.is_worker()
      • DistContext.is_server()
      • DistContext.is_client()
      • DistContext.num_servers()
      • DistContext.num_clients()
      • DistContext.worker_name
    • get_context()
    • init_worker_group()
  • dist_dataset
    • DistDataset
      • DistDataset.load()
      • DistDataset.share_ipc()
      • DistDataset.from_ipc_handle()
      • DistDataset.node_feat_pb
      • DistDataset.edge_feat_pb
    • rebuild_dist_dataset()
    • reduce_dist_dataset()
  • dist_feature
    • RpcFeatureLookupCallee
      • RpcFeatureLookupCallee.call()
    • DistFeature
      • DistFeature.local_get()
      • DistFeature.async_get()
  • dist_graph
    • DistGraph
      • DistGraph.get_local_graph()
      • DistGraph.get_node_partitions()
      • DistGraph.get_edge_partitions()
  • dist_neighbor_loader
    • DistNeighborLoader
  • dist_neighbor_sampler
    • PartialNeighborOutput
      • PartialNeighborOutput.index
      • PartialNeighborOutput.output
    • RpcSamplingCallee
      • RpcSamplingCallee.call()
    • RpcSubGraphCallee
      • RpcSubGraphCallee.call()
    • DistNeighborSampler
      • DistNeighborSampler.sample_from_nodes()
      • DistNeighborSampler.sample_from_edges()
      • DistNeighborSampler.subgraph()
  • dist_options
    • CollocatedDistSamplingWorkerOptions
    • MpDistSamplingWorkerOptions
    • RemoteDistSamplingWorkerOptions
  • dist_sampling_producer
    • MP_STATUS_CHECK_INTERVAL
    • MpCommand
      • MpCommand.SAMPLE_ALL
      • MpCommand.STOP
    • DistMpSamplingProducer
      • DistMpSamplingProducer.init()
      • DistMpSamplingProducer.shutdown()
      • DistMpSamplingProducer.produce_all()
    • DistCollocatedSamplingProducer
      • DistCollocatedSamplingProducer.init()
      • DistCollocatedSamplingProducer.shutdown()
      • DistCollocatedSamplingProducer.reset()
      • DistCollocatedSamplingProducer.sample()
  • dist_server
    • SERVER_EXIT_STATUS_CHECK_INTERVAL
    • DistServer
      • DistServer.shutdown()
      • DistServer.wait_for_exit()
      • DistServer.exit()
      • DistServer.get_dataset_meta()
      • DistServer.create_sampling_producer()
      • DistServer.destroy_sampling_producer()
      • DistServer.start_new_epoch_sampling()
      • DistServer.fetch_one_sampled_message()
    • get_server()
    • init_server()
    • wait_and_shutdown_server()
  • event_loop
    • wrap_torch_future()
    • ConcurrentEventLoop
      • ConcurrentEventLoop.start_loop()
      • ConcurrentEventLoop.shutdown_loop()
      • ConcurrentEventLoop.wait_all()
      • ConcurrentEventLoop.add_task()
      • ConcurrentEventLoop.run_task()
  • rpc
    • rpc_is_initialized()
    • get_rpc_master_addr()
    • get_rpc_master_port()
    • get_rpc_current_group_worker_names()
    • get_rpc_worker_names()
    • all_gather()
    • barrier()
    • global_all_gather()
    • global_barrier()
    • init_rpc()
    • shutdown_rpc()
    • RpcDataPartitionRouter
      • RpcDataPartitionRouter.get_to_worker()
    • rpc_sync_data_partitions()
    • RpcCalleeBase
      • RpcCalleeBase.call()
    • rpc_register()
    • rpc_request_async()
    • rpc_request()
    • rpc_global_request_async()
    • rpc_global_request()
• graphlearn_torch.loader
  • neighbor_loader
    • NeighborLoader
  • transform
    • to_data()
    • to_hetero_data()
• graphlearn_torch.partition
  • feature_partition
  • graph_partition
• graphlearn_torch.sampler
  • base
    • EdgeIndex
      • EdgeIndex.edge_index
      • EdgeIndex.e_id
      • EdgeIndex.size
      • EdgeIndex.to()
    • NodeSamplerInput
      • NodeSamplerInput.node
      • NodeSamplerInput.input_type
      • NodeSamplerInput.share_memory()
      • NodeSamplerInput.to()
    • NegativeSamplingMode
      • NegativeSamplingMode.binary
      • NegativeSamplingMode.triplet
    • NegativeSampling
      • NegativeSampling.mode
      • NegativeSampling.weight
      • NegativeSampling.amount
      • NegativeSampling.is_binary()
      • NegativeSampling.is_triplet()
      • NegativeSampling.share_memory()
      • NegativeSampling.to()
    • EdgeSamplerInput
      • EdgeSamplerInput.row
      • EdgeSamplerInput.col
      • EdgeSamplerInput.label
      • EdgeSamplerInput.input_type
      • EdgeSamplerInput.neg_sampling
      • EdgeSamplerInput.share_memory()
      • EdgeSamplerInput.to()
    • SamplerOutput
      • SamplerOutput.node
      • SamplerOutput.row
      • SamplerOutput.col
      • SamplerOutput.edge
      • SamplerOutput.batch
      • SamplerOutput.device
      • SamplerOutput.metadata
    • HeteroSamplerOutput
      • HeteroSamplerOutput.node
      • HeteroSamplerOutput.row
      • HeteroSamplerOutput.col
      • HeteroSamplerOutput.edge
      • HeteroSamplerOutput.batch
      • HeteroSamplerOutput.edge_types
      • HeteroSamplerOutput.input_type
      • HeteroSamplerOutput.device
      • HeteroSamplerOutput.metadata
      • HeteroSamplerOutput.get_edge_index()
    • NeighborOutput
      • NeighborOutput.nbr
      • NeighborOutput.nbr_num
      • NeighborOutput.edge
      • NeighborOutput.to()
    • SamplingType
      • SamplingType.NODE
      • SamplingType.LINK
      • SamplingType.SUBGRAPH
      • SamplingType.RANDOM_WALK
    • SamplingConfig
      • SamplingConfig.sampling_type
      • SamplingConfig.num_neighbors
      • SamplingConfig.batch_size
      • SamplingConfig.shuffle
      • SamplingConfig.drop_last
      • SamplingConfig.with_edge
      • SamplingConfig.collect_features
      • SamplingConfig.with_neg
    • BaseSampler
      • BaseSampler.sample_from_nodes()
      • BaseSampler.sample_from_edges()
      • BaseSampler.subgraph()
  • negative_sampler
    • RandomNegativeSampler
      • RandomNegativeSampler.sample()
  • neighbor_sampler
    • NeighborSampler
      • NeighborSampler.subgraph_op
      • NeighborSampler.lazy_init_sampler()
      • NeighborSampler.lazy_init_neg_sampler()
      • NeighborSampler.lazy_init_subgraph_op()
      • NeighborSampler.sample_one_hop()
      • NeighborSampler.sample_from_nodes()
      • NeighborSampler.sample_from_edges()
      • NeighborSampler.sample_pyg_v1()
      • NeighborSampler.subgraph()
      • NeighborSampler.sample_prob()
      • NeighborSampler.get_inducer()
      • NeighborSampler.create_inducer()
• graphlearn_torch.utils
  • common
    • ensure_dir()
    • merge_dict()
    • get_free_port()
    • id2idx_v2()
    • index_select()
    • merge_hetero_sampler_output()
    • format_hetero_sampler_output()
  • device
    • get_available_device()
    • assign_device()
    • ensure_device()
  • exit_status
    • python_exit_status
  • mixin
    • CastMixin
      • CastMixin.cast()
  • singleton
    • singleton()
  • tensor
    • tensor_equal_with_device()
    • id2idx()
    • convert_to_tensor()
    • apply_to_all_tensor()
    • share_memory()
    • squeeze()
  • units
    • parse_size()