Index

_

__add__() (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __call__() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) __cat_dim__() (src.common.graph_builder.pyg_graph_data.PygGraphData method) __class_getitem__() (common.collections.frozen_dict.FrozenDict class method) (common.collections.sorted_dict.SortedDict class method) (common.types.uri.local_uri.LocalUri class method) (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __contains__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __delattr__() (common.utils.test_utils.TestArgs method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) (src.common.types.graph_data.Edge method) (src.common.types.graph_data.Node method) (src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper method) (src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper method) (src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper method) (src.common.types.pb_wrappers.graph_data_types.NodePbWrapper method) __delitem__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) __dir__() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) (src.common.types.dataflow_job_options.CommonOptions method) __eq__() (common.collections.frozen_dict.FrozenDict method) (common.collections.sorted_dict.SortedDict method) (common.services.vertex_ai.VertexAiJobConfig method) (common.types.resource_config.CommonPipelineComponentConfigs method) (common.types.uri.gcs_uri.GcsUri method) (common.types.uri.http_uri.HttpUri method) (common.types.uri.local_uri.LocalUri method) (common.types.uri.uri.Uri method) (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) (common.utils.test_utils.TestArgs method) (orchestration.kubeflow.kfp_orchestrator.KfpEnvMetadata method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.types.graph_data.Edge method) (src.common.types.graph_data.EdgeType method) (src.common.types.graph_data.Node method) (src.common.types.model_eval_metrics.EvalMetric method) (src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper method) (src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper method) (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper method) (src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper method) (src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper method) (src.common.types.pb_wrappers.graph_data_types.NodePbWrapper method) (src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper method) (src.common.types.pb_wrappers.sampling_op.RandomUniformPbWrapper method) (src.common.types.pb_wrappers.sampling_op.RandomWeightedPbWrapper method) (src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) (src.common.types.pb_wrappers.sampling_op.TopKPbWrapper method) (src.common.types.pb_wrappers.sampling_op.UserDefinedPbWrapper method) (src.common.types.pb_wrappers.subgraph_sampling_strategy.GlobalRandomUniformStrategyPbWrapper method) (src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathPbWrapper method) (src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathStrategyPbWrapper method) (src.common.types.pb_wrappers.subgraph_sampling_strategy.SubgraphSamplingStrategyPbWrapper method) (src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper method) (src.common.types.pb_wrappers.trained_model_metadata.TrainedModelMetadataPbWrapper method) (src.common.utils.spark_job_manager.DataprocClusterInitData method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) (src.training.v1.lib.data_loaders.common.DataloaderConfig method) __fspath__() (common.types.uri.local_uri.LocalUri method) __ge__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __getattribute__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __getitem__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __getnewargs__() (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __gt__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.types.graph_data.EdgeType method) (src.common.types.graph_data.Node method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __hash__ (common.collections.sorted_dict.SortedDict attribute) (common.services.vertex_ai.VertexAiJobConfig attribute) (common.types.resource_config.CommonPipelineComponentConfigs attribute) (common.utils.compute.serialization.serialize_np.EncodedNdArray attribute) (orchestration.kubeflow.kfp_orchestrator.KfpEnvMetadata attribute) (src.common.graph_builder.pyg_graph_data.PygGraphData attribute) (src.common.types.model_eval_metrics.EvalMetric attribute) (src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper attribute) (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper attribute) (src.common.types.pb_wrappers.sampling_op.RandomUniformPbWrapper attribute) (src.common.types.pb_wrappers.sampling_op.RandomWeightedPbWrapper attribute) (src.common.types.pb_wrappers.sampling_op.TopKPbWrapper attribute) (src.common.types.pb_wrappers.sampling_op.UserDefinedPbWrapper attribute) (src.common.types.pb_wrappers.subgraph_sampling_strategy.GlobalRandomUniformStrategyPbWrapper attribute) (src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathPbWrapper attribute) (src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathStrategyPbWrapper attribute) (src.common.types.pb_wrappers.subgraph_sampling_strategy.SubgraphSamplingStrategyPbWrapper attribute) (src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper attribute) (src.common.types.pb_wrappers.trained_model_metadata.TrainedModelMetadataPbWrapper attribute) (src.common.utils.spark_job_manager.DataprocClusterInitData attribute) (src.training.v1.lib.data_loaders.common.DataloaderConfig attribute) __hash__() (common.collections.frozen_dict.FrozenDict method) (common.types.uri.gcs_uri.GcsUri method) (common.types.uri.http_uri.HttpUri method) (common.types.uri.local_uri.LocalUri method) (common.types.uri.uri.Uri method) (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) (common.utils.test_utils.TestArgs method) (src.common.types.graph_data.Edge method) (src.common.types.graph_data.EdgeType method) (src.common.types.graph_data.Node method) (src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper method) (src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper method) (src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper method) (src.common.types.pb_wrappers.graph_data_types.NodePbWrapper method) (src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper method) (src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __inc__() (src.common.graph_builder.pyg_graph_data.PygGraphData method) __init__() (common.collections.frozen_dict.FrozenDict method) (common.collections.sorted_dict.SortedDict method) (common.data.export.EmbeddingExporter method) (common.logger.Logger method) (common.metrics.base_metrics.NopMetricsPublisher method) (common.services.dataproc.DataprocService method) (common.services.gcp_cloud_logging.GCPCloudLoggingService method) (common.services.gcp_cloud_logging.LogEntries method) (common.services.kfp.KFPService method) (common.services.vertex_ai.VertexAiJobConfig method) (common.services.vertex_ai.VertexAIService method) (common.types.resource_config.CommonPipelineComponentConfigs method) (common.types.uri.gcs_uri.GcsUri method) (common.types.uri.http_uri.HttpUri method) (common.types.uri.local_uri.LocalUri method) (common.types.uri.uri.Uri method) (common.types.wrappers.argo_workflow_manifest.ArgoWorkflowManifestWrapper method) (common.types.wrappers.argo_workflow_manifest.ArgoWorkflowNodeStatus method) (common.types.wrappers.kfp_api.ApiRunDetailWrapper method) (common.utils.compute.serialization.coder.CoderProtocol method) (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (common.utils.compute.serialization.serialize_np.NumpyCoder method) (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) (common.utils.compute.serialization.serialize_string.StringCoder method) (common.utils.gcs.GcsUtils method) (common.utils.kfp.KfpOutputViewers method) (common.utils.proto_utils.ProtoUtils method) (common.utils.test_utils.TestArgs method) (orchestration.kubeflow.kfp_orchestrator.KfpEnvMetadata method) (orchestration.kubeflow.kfp_orchestrator.KfpOrchestrator method) (src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol method) (src.common.graph_builder.pyg_graph_builder.PygGraphBuilder method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.modeling_task_specs.utils.early_stop.EarlyStopper method) (src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler method) (src.common.models.layers.count_min_sketch.CountMinSketch method) (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) (src.common.types.dataflow_job_options.CommonOptions method) (src.common.types.graph_data.Edge method) (src.common.types.graph_data.Node method) (src.common.types.model_eval_metrics.EvalMetric method) (src.common.types.model_eval_metrics.EvalMetricsCollection method) (src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper method) (src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper method) (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper method) (src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper method) (src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper method) (src.common.types.pb_wrappers.graph_data_types.NodePbWrapper method) (src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper method) (src.common.types.pb_wrappers.sampling_op.RandomUniformPbWrapper method) (src.common.types.pb_wrappers.sampling_op.RandomWeightedPbWrapper method) (src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) (src.common.types.pb_wrappers.sampling_op.TopKPbWrapper method) (src.common.types.pb_wrappers.sampling_op.UserDefinedPbWrapper method) (src.common.types.pb_wrappers.subgraph_sampling_strategy.GlobalRandomUniformStrategyPbWrapper method) (src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathPbWrapper method) (src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathStrategyPbWrapper method) (src.common.types.pb_wrappers.subgraph_sampling_strategy.SubgraphSamplingStrategyPbWrapper method) (src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper method) (src.common.types.pb_wrappers.trained_model_metadata.TrainedModelMetadataPbWrapper method) (src.common.utils.bq.BqUtils method) (src.common.utils.file_loader.FileLoader method) (src.common.utils.scheduler.CosineDecayScheduler method) (src.common.utils.spark_job_manager.DataprocClusterInitData method) (src.common.utils.spark_job_manager.SparkJobManager method) (src.config_populator.config_populator.ConfigPopulator method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchemaBuilder method) (src.training.v1.lib.data_loaders.common.DataloaderConfig method)

__init_subclass__() (common.collections.frozen_dict.FrozenDict class method) (common.collections.sorted_dict.SortedDict class method) (common.utils.compute.serialization.coder.CoderProtocol class method) (common.utils.compute.serialization.serialize_np.NumpyCoder class method) (common.utils.compute.serialization.serialize_string.StringCoder class method) (src.common.graph_builder.abstract_graph_builder.GraphBuilder class method) (src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol class method) (src.common.graph_builder.pyg_graph_builder.PygGraphBuilder class method) (src.common.graph_builder.pyg_graph_data.PygGraphData class method) __ior__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) __iter__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __le__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __len__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __lt__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.types.graph_data.EdgeType method) (src.common.types.graph_data.Node method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __mul__() (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __ne__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __new__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.types.graph_data.EdgeType static method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema static method) __or__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) __post_init__() (src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathPbWrapper method) __reduce__() (src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) __repr__() (common.collections.frozen_dict.FrozenDict method) (common.collections.sorted_dict.SortedDict method) (common.logger.Logger method) (common.services.vertex_ai.VertexAiJobConfig method) (common.types.resource_config.CommonPipelineComponentConfigs method) (common.types.uri.gcs_uri.GcsUri method) (common.types.uri.http_uri.HttpUri method) (common.types.uri.local_uri.LocalUri method) (common.types.uri.uri.Uri method) (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) (common.utils.test_utils.TestArgs method) (orchestration.kubeflow.kfp_orchestrator.KfpEnvMetadata method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) (src.common.types.graph_data.Edge method) (src.common.types.graph_data.EdgeType method) (src.common.types.graph_data.Node method) (src.common.types.model_eval_metrics.EvalMetric method) (src.common.types.model_eval_metrics.EvalMetricsCollection method) (src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper method) (src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper method) (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper method) (src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper method) (src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper method) (src.common.types.pb_wrappers.graph_data_types.NodePbWrapper method) (src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper method) (src.common.types.pb_wrappers.sampling_op.RandomUniformPbWrapper method) (src.common.types.pb_wrappers.sampling_op.RandomWeightedPbWrapper method) (src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) (src.common.types.pb_wrappers.sampling_op.TopKPbWrapper method) (src.common.types.pb_wrappers.sampling_op.UserDefinedPbWrapper method) (src.common.types.pb_wrappers.subgraph_sampling_strategy.GlobalRandomUniformStrategyPbWrapper method) (src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathPbWrapper method) (src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathStrategyPbWrapper method) (src.common.types.pb_wrappers.subgraph_sampling_strategy.SubgraphSamplingStrategyPbWrapper method) (src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper method) (src.common.types.pb_wrappers.trained_model_metadata.TrainedModelMetadataPbWrapper method) (src.common.utils.spark_job_manager.DataprocClusterInitData method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) (src.training.v1.lib.data_loaders.common.DataloaderConfig method) __reversed__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) __rmul__() (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) __ror__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) __setattr__() (common.utils.test_utils.TestArgs method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) (src.common.types.dataflow_job_options.CommonOptions method) (src.common.types.graph_data.Edge method) (src.common.types.graph_data.Node method) (src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper method) (src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper method) (src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper method) (src.common.types.pb_wrappers.graph_data_types.NodePbWrapper method) __setitem__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) __sizeof__() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) __str__() (src.common.types.dataflow_job_options.CommonOptions method) __subclasshook__() (common.collections.frozen_dict.FrozenDict class method) (common.collections.sorted_dict.SortedDict class method) (common.types.uri.local_uri.LocalUri class method) (common.utils.compute.serialization.coder.CoderProtocol method) (common.utils.compute.serialization.serialize_np.NumpyCoder method) (common.utils.compute.serialization.serialize_string.StringCoder method) (src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) __weakref__ (common.collections.frozen_dict.FrozenDict attribute) (common.collections.sorted_dict.SortedDict attribute) (common.data.export.EmbeddingExporter attribute) (common.logger.Logger attribute) (common.metrics.base_metrics.NopMetricsPublisher attribute) (common.metrics.metrics_interface.OpsMetricPublisher attribute) (common.services.dataproc.DataprocService attribute) (common.services.gcp_cloud_logging.GCPCloudLoggingService attribute) (common.services.gcp_cloud_logging.LogEntries attribute) (common.services.kfp.KFPService attribute) (common.services.vertex_ai.VertexAiJobConfig attribute) (common.services.vertex_ai.VertexAIService attribute) (common.types.resource_config.CommonPipelineComponentConfigs attribute) (common.types.uri.gcs_uri.GcsUri attribute) (common.types.uri.http_uri.HttpUri attribute) (common.types.uri.local_uri.LocalUri attribute) (common.types.uri.uri.Uri attribute) (common.types.uri.uri_factory.UriFactory attribute) (common.types.wrappers.argo_workflow_manifest.ArgoWorkflowManifestWrapper attribute) (common.types.wrappers.argo_workflow_manifest.ArgoWorkflowNodeStatus attribute) (common.types.wrappers.kfp_api.ApiRunDetailWrapper attribute) (common.utils.compute.serialization.coder.CoderProtocol attribute) (common.utils.compute.serialization.serialize_np.EncodedNdArray attribute) (common.utils.compute.serialization.serialize_np.NumpyCoder attribute) (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder attribute) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder attribute) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder attribute) (common.utils.compute.serialization.serialize_string.StringCoder attribute) (common.utils.gcs.GcsUtils attribute) (common.utils.kfp.KfpOutputViewers attribute) (common.utils.proto_utils.ProtoUtils attribute) (common.utils.test_utils.TestArgs attribute) (orchestration.kubeflow.kfp_orchestrator.KfpEnvMetadata attribute) (orchestration.kubeflow.kfp_orchestrator.KfpOrchestrator attribute) (src.common.graph_builder.abstract_graph_builder.GraphBuilder attribute) (src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol attribute) (src.common.graph_builder.pyg_graph_builder.PygGraphBuilder attribute) (src.common.graph_builder.pyg_graph_data.PygGraphData attribute) (src.common.modeling_task_specs.utils.early_stop.EarlyStopper attribute) (src.common.modeling_task_specs.utils.parameter_loading.ArgumentReader attribute) (src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler attribute) (src.common.models.layers.count_min_sketch.CountMinSketch attribute) (src.common.models.layers.decoder.LinkPredictionDecoder attribute) (src.common.models.layers.feature_interaction.DCNCross attribute) (src.common.models.layers.feature_interaction.DCNv2 attribute) (src.common.models.pyg.heterogeneous.HGT attribute) (src.common.models.pyg.heterogeneous.SimpleHGN attribute) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv attribute) (src.common.models.pyg.nn.conv.gin_conv.GINEConv attribute) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction attribute) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge attribute) (src.common.translators.gbml_protos_translator.GbmlProtosTranslator attribute) (src.common.translators.model_eval_metrics_translator.EvalMetricsCollectionTranslator attribute) (src.common.types.dataflow_job_options.CommonOptions attribute) (src.common.types.graph_data.Edge attribute) (src.common.types.graph_data.EdgeUsageType attribute) (src.common.types.graph_data.Node attribute) (src.common.types.model_eval_metrics.EvalMetric attribute) (src.common.types.model_eval_metrics.EvalMetricsCollection attribute) (src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper attribute) (src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper attribute) (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper attribute) (src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper attribute) (src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper attribute) (src.common.types.pb_wrappers.graph_data_types.NodePbWrapper attribute) (src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper attribute) (src.common.types.pb_wrappers.sampling_op.RandomUniformPbWrapper attribute) (src.common.types.pb_wrappers.sampling_op.RandomWeightedPbWrapper attribute) (src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper attribute) (src.common.types.pb_wrappers.sampling_op.TopKPbWrapper attribute) (src.common.types.pb_wrappers.sampling_op.UserDefinedPbWrapper attribute) (src.common.types.pb_wrappers.subgraph_sampling_strategy.GlobalRandomUniformStrategyPbWrapper attribute) (src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathPbWrapper attribute) (src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathStrategyPbWrapper attribute) (src.common.types.pb_wrappers.subgraph_sampling_strategy.SubgraphSamplingStrategyPbWrapper attribute) (src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper attribute) (src.common.types.pb_wrappers.trained_model_metadata.TrainedModelMetadataPbWrapper attribute) (src.common.types.task_metadata.TaskMetadataType attribute) (src.common.utils.bq.BqUtils attribute) (src.common.utils.data.feature_serialization.FeatureSerializationUtils attribute) (src.common.utils.file_loader.FileLoader attribute) (src.common.utils.gbml_protos.TrainingSamplesSchemaProtoUtils attribute) (src.common.utils.scheduler.CosineDecayScheduler attribute) (src.common.utils.spark_job_manager.DataprocClusterInitData attribute) (src.common.utils.spark_job_manager.SparkJobManager attribute) (src.config_populator.config_populator.ConfigPopulator attribute) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchemaBuilder attribute) (src.post_process.lib.base_post_processor.BasePostProcessor attribute) (src.training.v1.lib.data_loaders.common.DataloaderConfig attribute)

A

absolute() (common.types.uri.local_uri.LocalUri method) add() (src.common.models.layers.count_min_sketch.CountMinSketch method) add_child_sampling_op_pb_wrapper() (src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) add_edge() (src.common.graph_builder.abstract_graph_builder.GraphBuilder method) (src.common.graph_builder.pyg_graph_builder.PygGraphBuilder method) add_embedding() (common.data.export.EmbeddingExporter method) add_graph_data() (src.common.graph_builder.abstract_graph_builder.GraphBuilder method) (src.common.graph_builder.pyg_graph_builder.PygGraphBuilder method) add_module() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) add_node() (src.common.graph_builder.abstract_graph_builder.GraphBuilder method) (src.common.graph_builder.pyg_graph_builder.PygGraphBuilder method) add_parent_sampling_op_pb_wrapper() (src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) add_task_resource_requirements() (in module orchestration.kubeflow.utils.resource) add_torch_long_tensor() (src.common.models.layers.count_min_sketch.CountMinSketch method) aggregate() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)

ApiRunDetailWrapper (class in common.types.wrappers.kfp_api) append_line_to_file() (in module common.utils.local_fs) apply() (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) apply_() (src.common.graph_builder.pyg_graph_data.PygGraphData method) are_disjoint() (src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol static method) (src.common.graph_builder.pyg_graph_data.PygGraphData static method) are_same_graph() (src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol static method) (src.common.graph_builder.pyg_graph_data.PygGraphData static method) ArgoWorkflowManifestWrapper (class in common.types.wrappers.argo_workflow_manifest) ArgoWorkflowNodeStatus (class in common.types.wrappers.argo_workflow_manifest) ArgumentReader (class in src.common.modeling_task_specs.utils.parameter_loading) as_deterministic_coder() (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) assert_component_runtimes_match_expected_parameters() (in module src.post_process.utils.component_runtime) assert_cosine_similarity_stats() (in module src.post_process.utils.cosine_similarity) assert_proto_field_value_is_truthy() (in module src.validation_check.libs.utils) assert_proto_has_field() (in module src.validation_check.libs.utils)

B

BasePostProcessor (class in src.post_process.lib.base_post_processor) batch() (in module common.collections.itertools) bfloat16() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)

BqUtils (class in src.common.utils.bq) buffers() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) build() (src.common.graph_builder.abstract_graph_builder.GraphBuilder method) (src.common.graph_builder.pyg_graph_builder.PygGraphBuilder method)

C

calculate_cosine_sim_between_embedding_tables() (in module src.post_process.utils.cosine_similarity) calculate_cosine_similarity_stats() (in module src.post_process.utils.cosine_similarity) calculate_in_batch_candidate_sampling_probability() (in module src.common.models.layers.count_min_sketch) check_glt_backend_eligibility_component() (in module orchestration.kubeflow.utils.glt_backend) check_if_dag_contains_cycles() (src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) check_if_inferencer_resource_config_valid() (in module src.validation_check.libs.resource_config_checks) check_if_preprocessor_resource_config_valid() (in module src.validation_check.libs.resource_config_checks) check_if_shared_resource_config_valid() (in module src.validation_check.libs.resource_config_checks) check_if_split_generator_resource_config_valid() (in module src.validation_check.libs.resource_config_checks) check_if_subgraph_sampler_resource_config_valid() (in module src.validation_check.libs.resource_config_checks) check_if_trainer_resource_config_valid() (in module src.validation_check.libs.resource_config_checks) check_sampling_op_edge_type_validity() (src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) child_sampling_op_pb_wrappers (src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper property) children() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) clear() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) clone() (src.common.graph_builder.pyg_graph_data.PygGraphData method) coalesce() (src.common.graph_builder.pyg_graph_data.PygGraphData method) CoderProtocol (class in common.utils.compute.serialization.coder) collect() (src.common.graph_builder.pyg_graph_data.PygGraphData method) CombinationMode (class in src.common.models.pyg.nn.models.feature_interaction) common module common.beam module common.collections module common.collections.frozen_dict module common.collections.itertools module common.collections.sorted_dict module common.data module common.data.export module common.env_config module common.logger module common.metrics module common.metrics.base_metrics module common.metrics.decorators module common.metrics.metrics_interface module common.services module common.services.dataproc module common.services.gcp_cloud_logging module common.services.kfp module common.services.vertex_ai module common.types module common.types.resource_config module common.types.uri module common.types.uri.gcs_uri module common.types.uri.http_uri module common.types.uri.local_uri module common.types.uri.uri module common.types.uri.uri_factory module common.types.wrappers module common.types.wrappers.argo_workflow_manifest module common.types.wrappers.kfp_api module common.utils module common.utils.compute module common.utils.compute.random module common.utils.compute.serialization module

common.utils.compute.serialization.coder module common.utils.compute.serialization.serialize_np module common.utils.compute.serialization.serialize_protos module common.utils.compute.serialization.serialize_string module common.utils.dataflow module common.utils.func_tools module common.utils.gcs module common.utils.kfp module common.utils.local_fs module common.utils.os_utils module common.utils.proto_utils module common.utils.retry module common.utils.test_utils module common.utils.torch_training module CommonOptions (class in src.common.types.dataflow_job_options) CommonPipelineComponentConfigs (class in common.types.resource_config) compile() (orchestration.kubeflow.kfp_orchestrator.KfpOrchestrator class method) (orchestration.kubeflow.kfp_orchestrator.KfpOrchestrator method) (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) component_status_by_component_display_name (common.types.wrappers.argo_workflow_manifest.ArgoWorkflowManifestWrapper property) concat() (src.common.graph_builder.pyg_graph_data.PygGraphData method) condensed_edge_type_to_condensed_node_types (src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) ConfigPopulator (class in src.config_populator.config_populator) connect_worker_pool() (common.services.vertex_ai.VertexAIService static method) contiguous() (src.common.graph_builder.pyg_graph_data.PygGraphData method) convert_days_to_ms() (in module src.common.utils.time) coo() (src.common.graph_builder.pyg_graph_data.PygGraphData method) copy() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) copy_files() (in module common.utils.local_fs) CosineDecayScheduler (class in src.common.utils.scheduler) count() (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) count_files_with_uri_prefix() (in module common.utils.local_fs) CountMinSketch (class in src.common.models.layers.count_min_sketch) cpu() (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) create_cluster() (common.services.dataproc.DataprocService method) create_config_populator_task_op() (in module orchestration.kubeflow.kfp_pipeline) create_data_preprocessor_task_op() (in module orchestration.kubeflow.kfp_pipeline) create_empty_file_if_none_exists() (in module common.utils.local_fs) create_file_symlinks() (in module common.utils.local_fs) create_inferencer_task_op() (in module orchestration.kubeflow.kfp_pipeline) create_post_processor_task_op() (in module orchestration.kubeflow.kfp_pipeline) create_split_generator_task_op() (in module orchestration.kubeflow.kfp_pipeline) create_subgraph_sampler_task_op() (in module orchestration.kubeflow.kfp_pipeline) create_trainer_task_op() (in module orchestration.kubeflow.kfp_pipeline) create_uri() (common.types.uri.uri_factory.UriFactory static method) critical() (common.logger.Logger method) csc() (src.common.graph_builder.pyg_graph_data.PygGraphData method) csr() (src.common.graph_builder.pyg_graph_data.PygGraphData method) cuda() (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) current_datetime() (in module src.common.utils.time) current_formatted_datetime() (in module src.common.utils.time)

D

dataflow_runner (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) DataloaderConfig (class in src.training.v1.lib.data_loaders.common) DataloaderTypes (class in src.training.v1.lib.data_loaders.common) DataprocClusterInitData (class in src.common.utils.spark_job_manager) DataprocService (class in common.services.dataproc) dataset_type (src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper property) DatasetMetadataPbWrapper (class in src.common.types.pb_wrappers.dataset_metadata) DatasetSplit (class in src.common.types.dataset_split) DCNCross (class in src.common.models.layers.feature_interaction) DCNv2 (class in src.common.models.layers.feature_interaction) debug() (common.logger.Logger method) decode() (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) decode_nested() (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) DecoderType (class in src.common.models.layers.decoder) delete_and_create_local_path() (in module common.utils.local_fs) delete_bq_table_if_exist() (src.common.utils.bq.BqUtils method) delete_cluster() (common.services.dataproc.DataprocService method)

delete_files() (src.common.utils.file_loader.FileLoader method) delete_local_directory() (in module common.utils.local_fs) detach() (src.common.graph_builder.pyg_graph_data.PygGraphData method) detach_() (src.common.graph_builder.pyg_graph_data.PygGraphData method) display_data() (src.common.types.dataflow_job_options.CommonOptions method) does_cluster_exist() (common.services.dataproc.DataprocService method) does_path_exist() (in module common.utils.local_fs) does_uri_exist() (src.common.utils.file_loader.FileLoader method) double() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) download_files_from_gcs_paths_to_local_paths() (common.utils.gcs.GcsUtils method) dst_node_type (src.common.types.graph_data.EdgeType attribute)

E

EarlyStopper (class in src.common.modeling_task_specs.utils.early_stop) Edge (class in src.common.types.graph_data) edge_attrs() (src.common.graph_builder.pyg_graph_data.PygGraphData method) edge_items() (src.common.graph_builder.pyg_graph_data.PygGraphData method) edge_stores (src.common.graph_builder.pyg_graph_data.PygGraphData property) edge_subgraph() (src.common.graph_builder.pyg_graph_data.PygGraphData method) edge_type_subgraph() (src.common.graph_builder.pyg_graph_data.PygGraphData method) edge_types (src.common.graph_builder.pyg_graph_data.PygGraphData property) edge_types_to_be_registered (src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol property) (src.common.graph_builder.pyg_graph_data.PygGraphData property) edge_update() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) edge_updater() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) EdgeAttrGATConv (class in src.common.models.pyg.nn.conv.edge_attr_gat_conv) EdgePbWrapper (class in src.common.types.pb_wrappers.graph_data_types) EdgePbWrapperCoder (class in common.utils.compute.serialization.serialize_protos) EdgeType (class in src.common.types.graph_data) EdgeUsageType (class in src.common.types.graph_data) embedding_bq_dataset_name (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) EmbeddingExporter (class in common.data.export) encode() (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) encode_nested() (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) EncodedNdArray (class in common.utils.compute.serialization.serialize_np) env module

env.dep_constants module env.pipelines_config module error() (common.logger.Logger method) estimate() (src.common.models.layers.count_min_sketch.CountMinSketch method) estimate_size() (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) estimate_torch_long_tensor() (src.common.models.layers.count_min_sketch.CountMinSketch method) eval() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) EvalMetric (class in src.common.types.model_eval_metrics) EvalMetricsCollection (class in src.common.types.model_eval_metrics) EvalMetricsCollectionTranslator (class in src.common.translators.model_eval_metrics_translator) EvalMetricType (class in src.common.types.model_eval_metrics) exception() (common.logger.Logger method) extra_repr() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)

F

FeatureInteraction (class in src.common.models.pyg.nn.models.feature_interaction) FeatureSerializationUtils (class in src.common.utils.data.feature_serialization) FeatureTypes (class in src.common.types.features) fetch_bq_table_schema() (src.common.utils.bq.BqUtils method) FileLoader (class in src.common.utils.file_loader) FileSystemEntity (class in common.utils.local_fs) filter_dict() (in module src.common.models.pyg.utils) FlattenedGraphMetadataPbWrapper (class in src.common.types.pb_wrappers.flattened_graph_metadata) float() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) flush_embeddings() (common.data.export.EmbeddingExporter method)

flushes_metrics() (in module common.metrics.decorators) format_bq_path() (src.common.utils.bq.BqUtils static method) format_datetime() (in module src.common.utils.time) forward() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) from_dict() (src.common.graph_builder.pyg_graph_data.PygGraphData class method) from_dictionary() (src.common.types.dataflow_job_options.CommonOptions class method) from_runner_api() (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder class method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder class method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder class method) fromkeys() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) FrozenDict (class in common.collections.frozen_dict)

G

GbmlGraphDataProtocol (class in src.common.graph_builder.gbml_graph_protocol) GbmlProtosTranslator (class in src.common.translators.gbml_protos_translator) GCPCloudLoggingService (class in common.services.gcp_cloud_logging) GcsUri (class in common.types.uri.gcs_uri) GcsUtils (class in common.utils.gcs) generate_component_task() (in module orchestration.kubeflow.kfp_pipeline) generate_ids() (src.common.graph_builder.pyg_graph_data.PygGraphData method) generate_pipeline() (in module orchestration.kubeflow.kfp_pipeline) get() (common.collections.frozen_dict.FrozenDict method) (common.collections.sorted_dict.SortedDict method) (common.utils.compute.serialization.serialize_np.EncodedNdArray method) get_all_edge_attrs() (src.common.graph_builder.pyg_graph_data.PygGraphData method) get_all_options() (src.common.types.dataflow_job_options.CommonOptions method) get_all_tensor_attrs() (src.common.graph_builder.pyg_graph_data.PygGraphData method) get_applied_task_perm_gcs_path() (in module src.common.constants.gcs) get_applied_task_temp_gcs_path() (in module src.common.constants.gcs) get_applied_task_temp_regional_gcs_path() (in module src.common.constants.gcs) get_available_cpus() (in module common.env_config) get_buffer() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) get_compiled_jar_path() (in module env.dep_constants) get_config_populator_assets_perm_gcs_path() (in module src.common.constants.gcs) get_console_uri_from_pipeline_result() (in module common.utils.dataflow) get_current_jar_file() (in module env.dep_constants) get_data_preprocessor_assets_perm_gcs_path() (in module src.common.constants.gcs) get_data_preprocessor_assets_temp_gcs_path() (in module src.common.constants.gcs) get_data_preprocessor_staging_gcs_path() (in module src.common.constants.gcs) get_dataflow_staging_gcs_path() (in module src.common.constants.gcs) get_dataflow_temp_gcs_path() (in module src.common.constants.gcs) get_dehydrated_node_pb_wrappers_from_edge_wrapper() (in module src.common.types.pb_wrappers.graph_data_types_utils) get_distributed_backend() (in module common.utils.torch_training) get_edge_index() (src.common.graph_builder.pyg_graph_data.PygGraphData method) get_edge_store() (src.common.graph_builder.pyg_graph_data.PygGraphData method) get_embeddings_dataset_bq_path() (in module src.common.constants.bq) get_embeddings_table() (in module src.common.constants.bq) get_example_task_edge_features_gcs_dir() (in module src.common.constants.test_assets) get_example_task_edge_features_schema_gcs_path() (in module src.common.constants.test_assets) get_example_task_edges_bq_table_path() (in module src.common.constants.test_assets) get_example_task_frozen_gbml_config_gcs_path() (in module src.common.constants.test_assets) get_example_task_node_features_gcs_dir() (in module src.common.constants.test_assets) get_example_task_node_features_schema_gcs_path() (in module src.common.constants.test_assets) get_example_task_nodes_bq_table_path() (in module src.common.constants.test_assets) get_example_task_preprocess_gcs_prefix() (in module src.common.constants.test_assets) get_example_task_static_assets_gcs_dir() (in module src.common.constants.test_assets) get_experiment_name_by_id() (common.services.kfp.KFPService method) get_extra_state() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) get_file_name() (common.types.uri.local_uri.LocalUri method) get_frozen_gbml_config_proto_gcs_path() (in module src.common.constants.gcs) get_gbml_assets_tmp_path() (in module src.common.constants.local_fs) get_gbml_log_file_prefix_tmp_path() (in module src.common.constants.local_fs) get_gbml_logs_folder_local_tmp_path() (in module src.common.constants.local_fs) get_gbml_task_local_tmp_path() (in module src.common.constants.local_fs) get_gigl_root_directory() (in module src.common.constants.local_fs) get_global_edge_features_dict() (src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) get_global_node_features_dict() (src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) get_host_gcp_project_name() (common.services.kfp.KFPService method) get_host_k8_cluster_name() (common.services.kfp.KFPService method) get_host_name() (common.services.vertex_ai.VertexAIService static method) get_impl() (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) get_inference_embeddings_local_tmp_dir_path() (in module src.common.constants.local_fs) get_inference_embeddings_local_tmp_file_prefix() (in module src.common.constants.local_fs) get_inference_predictions_local_tmp_file_prefix() (in module src.common.constants.local_fs) get_inferencer_asset_dir_gcs_path() (in module src.common.constants.gcs) get_inferencer_embeddings_gcs_prefix() (in module src.common.constants.gcs) get_inferencer_predictions_gcs_prefix() (in module src.common.constants.gcs) get_jar_file_uri() (in module env.dep_constants) get_latest_experiment_from_name() (common.services.kfp.KFPService method) get_latest_run_with_name() (common.services.kfp.KFPService method) get_leader_hostname() (common.services.vertex_ai.VertexAIService static method) get_leader_port() (common.services.vertex_ai.VertexAIService static method) get_local_jar_directory() (in module env.dep_constants) get_logs_iterator_from_k8_container() (common.services.gcp_cloud_logging.GCPCloudLoggingService method) get_metric_name_for_gauge_model_metrics_hit_rate_at() (in module src.training.v1.lib.eval_metrics) get_metrics_service_instance() (in module src.common.utils.metrics_service_provider) get_node_store() (src.common.graph_builder.pyg_graph_data.PygGraphData method) get_output_paths() (src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper method) (src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper method)

get_parameter() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) get_path_to_manifest_file() (in module src.common.constants.local_fs) get_post_processor_asset_dir_gcs_path() (in module src.common.constants.gcs) get_post_processor_metrics_gcs_path() (in module src.common.constants.gcs) get_predictions_table() (in module src.common.constants.bq) get_preprocess_local_staging_dir_path() (in module src.common.constants.local_fs) get_preprocessed_metadata_proto_gcs_path() (in module src.common.constants.gcs) get_project_root_directory() (in module src.common.constants.local_fs) get_pyg_cora_dataset() (in module src.applied_tasks.test_tasks.academic) get_python_project_root_path() (in module src.common.constants.local_fs) get_rank() (common.services.vertex_ai.VertexAIService static method) (in module common.utils.torch_training) get_resource_config() (in module env.pipelines_config) get_resource_labels() (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper method) get_run() (common.services.kfp.KFPService method) get_running_job_ids_on_cluster() (common.services.dataproc.DataprocService method) get_runtime_manifest_from_kfp_pipeline() (in module src.post_process.utils.component_runtime) get_sanitized_dataflow_job_name() (in module src.common.utils.dataflow) get_split_dataset_main_samples_gcs_file_prefix() (in module src.common.constants.gcs) get_split_dataset_output_gcs_file_prefix() (in module src.common.constants.gcs) get_split_dataset_random_negatives_gcs_file_prefix() (in module src.common.constants.gcs) get_split_generator_assets_temp_gcs_path() (in module src.common.constants.gcs) get_subgraph_sampler_flattened_graph_metadata_output_gcs_path() (in module src.common.constants.gcs) get_subgraph_sampler_node_anchor_based_link_prediction_random_negatives_samples_prefix() (in module src.common.constants.gcs) get_subgraph_sampler_node_anchor_based_link_prediction_samples_prefix() (in module src.common.constants.gcs) get_subgraph_sampler_node_anchor_based_link_prediction_task_dir() (in module src.common.constants.gcs) get_subgraph_sampler_node_neighborhood_samples_dir() (in module src.common.constants.gcs) get_subgraph_sampler_node_neighborhood_samples_path_prefix() (in module src.common.constants.gcs) get_subgraph_sampler_root_dir() (in module src.common.constants.gcs) get_subgraph_sampler_supervised_link_based_task_dir() (in module src.common.constants.gcs) get_subgraph_sampler_supervised_link_based_task_labeled_samples_prefix() (in module src.common.constants.gcs) get_subgraph_sampler_supervised_link_based_task_unlabeled_samples_prefix() (in module src.common.constants.gcs) get_subgraph_sampler_supervised_node_classification_labeled_samples_prefix() (in module src.common.constants.gcs) get_subgraph_sampler_supervised_node_classification_task_dir() (in module src.common.constants.gcs) get_subgraph_sampler_supervised_node_classification_unlabeled_samples_prefix() (in module src.common.constants.gcs) get_submitted_job_ids() (common.services.dataproc.DataprocService method) get_submodule() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) get_supervision_edge_node_types() (src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper method) get_supervision_edge_types() (src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper method) get_supervision_node_types() (src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper method) get_table() (src.common.models.layers.count_min_sketch.CountMinSketch method) get_table_names_within_date_range() (src.common.utils.bq.BqUtils method) get_table_paths_via_timedelta() (in module src.post_process.utils.cosine_similarity) get_task_root_node_types() (src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper method) get_tensor() (src.common.graph_builder.pyg_graph_data.PygGraphData method) get_tensor_size() (src.common.graph_builder.pyg_graph_data.PygGraphData method) get_tensorboard_logs_gcs_path() (in module src.common.constants.gcs) get_tf_transform_directory_path() (in module src.common.constants.gcs) get_tf_transform_raw_data_schema_file_path() (in module src.common.constants.gcs) get_tf_transform_stats_directory_path() (in module src.common.constants.gcs) get_tf_transform_stats_file_path() (in module src.common.constants.gcs) get_tf_transform_temp_directory_path() (in module src.common.constants.gcs) get_tf_transform_visualized_facets_file_path() (in module src.common.constants.gcs) get_tf_transformed_features_schema_path() (in module src.common.constants.gcs) get_tf_transformed_features_transform_fn_assets_directory_path() (in module src.common.constants.gcs) get_train_val_info_local_tmp_file_path() (in module src.common.constants.local_fs) get_trained_model_eval_metrics_gcs_path() (in module src.common.constants.gcs) get_trained_model_gcs_path() (in module src.common.constants.gcs) get_trained_model_metadata_proto_gcs_path() (in module src.common.constants.gcs) get_trained_models_dir_gcs_path() (in module src.common.constants.gcs) get_trained_scripted_model_gcs_path() (in module src.common.constants.gcs) get_trainer_asset_dir_gcs_path() (in module src.common.constants.gcs) get_transformed_features_directory_path() (in module src.common.constants.gcs) get_transformed_features_file_prefix() (in module src.common.constants.gcs) get_vertex_ai_job_id() (common.services.vertex_ai.VertexAIService static method) get_world_size() (common.services.vertex_ai.VertexAIService static method) (in module common.utils.torch_training) getEffectiveLevel() (common.logger.Logger method) GiGLComponents (class in src.common.constants.components) GiglResourceConfigWrapper (class in src.common.types.pb_wrappers.gigl_resource_config) GINEConv (class in src.common.models.pyg.nn.conv.gin_conv) global_node_to_subgraph_node_mapping (src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol property) (src.common.graph_builder.pyg_graph_data.PygGraphData property) GlobalRandomUniformStrategyPbWrapper (class in src.common.types.pb_wrappers.subgraph_sampling_strategy) GraphBuilder (class in src.common.graph_builder.abstract_graph_builder) GraphMetadataPbWrapper (class in src.common.types.pb_wrappers.graph_metadata) GraphPbWrapper (class in src.common.types.pb_wrappers.graph_data_types) GraphPbWrapperCoder (class in common.utils.compute.serialization.serialize_protos)

H

half() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)

has_isolated_nodes() (src.common.graph_builder.pyg_graph_data.PygGraphData method) has_self_loops() (src.common.graph_builder.pyg_graph_data.PygGraphData method) hasHandlers() (common.logger.Logger method) HGT (class in src.common.models.pyg.heterogeneous) HGTConv (class in src.common.models.pyg.nn.conv.hgt_conv) hit_rate_at_k() (in module src.common.utils.eval_metrics) HttpUri (class in common.types.uri.http_uri) hydrate_subgraph_features() (src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper static method)

I

import_obj() (in module common.utils.os_utils) index() (src.common.types.graph_data.EdgeType method) (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema method) InferenceOutputBigqueryTableSchema (class in src.inference.v1.lib.inference_output_schema) InferenceOutputBigqueryTableSchemaBuilder (class in src.inference.v1.lib.inference_output_schema) inferencer_config (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) info() (common.logger.Logger method) init_beam_pipeline_options() (in module src.common.utils.dataflow) init_metrics_publisher_grouping_for_job() (in module src.common.utils.metrics_service_provider) initialize_metrics() (in module src.common.utils.metrics_service_provider) ipu() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)

is_coalesced() (src.common.graph_builder.pyg_graph_data.PygGraphData method) is_cuda (src.common.graph_builder.pyg_graph_data.PygGraphData property) is_currently_running_in_vertex_ai_job() (common.services.vertex_ai.VertexAIService static method) is_datetime_str_format_valid() (in module src.common.utils.time) is_deterministic() (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) is_directed() (src.common.graph_builder.pyg_graph_data.PygGraphData method) is_distributed_available_and_initialized() (in module common.utils.torch_training) is_distributed_local_debug() (in module common.utils.torch_training) is_resource_config_loaded() (in module env.pipelines_config) is_sorted() (src.common.graph_builder.pyg_graph_data.PygGraphData method) is_sorted_by_time() (src.common.graph_builder.pyg_graph_data.PygGraphData method) is_undirected() (src.common.graph_builder.pyg_graph_data.PygGraphData method) is_valid() (common.types.uri.gcs_uri.GcsUri class method) (common.types.uri.http_uri.HttpUri class method) (common.types.uri.local_uri.LocalUri class method) (common.types.uri.uri.Uri class method) isEnabledFor() (common.logger.Logger method) items() (common.collections.frozen_dict.FrozenDict method) (common.collections.sorted_dict.SortedDict method) (common.utils.compute.serialization.serialize_np.EncodedNdArray method)

J

jittable() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)

join() (common.types.uri.gcs_uri.GcsUri class method) (common.types.uri.http_uri.HttpUri class method) (common.types.uri.local_uri.LocalUri class method) (common.types.uri.uri.Uri class method) JumpingKnowledge (class in src.common.models.pyg.nn.models.jumping_knowledge)

K

keys() (common.collections.frozen_dict.FrozenDict method) (common.collections.sorted_dict.SortedDict method) (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.graph_builder.pyg_graph_data.PygGraphData method)

KfpEnvMetadata (class in orchestration.kubeflow.kfp_orchestrator) KfpOrchestrator (class in orchestration.kubeflow.kfp_orchestrator) KfpOutputViewers (class in common.utils.kfp) KFPService (class in common.services.kfp)

L

l2_normalize_embeddings() (in module src.common.models.layers.normalization) LinkPredictionDecoder (class in src.common.models.layers.decoder) list_at_path() (in module common.utils.local_fs) list_children() (src.common.utils.file_loader.FileLoader method) list_uris_with_gcs_path_pattern() (common.utils.gcs.GcsUtils method) load_embeddings_to_bigquery() (in module common.data.export) load_file_to_bq() (src.common.utils.bq.BqUtils method) load_scripted_model_from_uri() (in module src.common.utils.model) load_state_dict() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)

load_state_dict_from_uri() (in module src.common.utils.model) LocalUri (class in common.types.uri.local_uri) log() (common.logger.Logger method) log_eval_metrics_to_ui() (in module orchestration.kubeflow.utils.log_metrics) log_stats_for_pyg_planetoid_dataset() (in module src.applied_tasks.test_tasks.academic) LogEntries (class in common.services.gcp_cloud_logging) Logger (class in common.logger) lru_cache() (in module common.utils.func_tools)

M

make_compute_deterministic_and_set_seed() (in module common.utils.compute.random) mean_reciprocal_rank() (in module src.common.utils.eval_metrics) message() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) message_and_aggregate() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) MessagePassingPathPbWrapper (class in src.common.types.pb_wrappers.subgraph_sampling_strategy) MessagePassingPathStrategyPbWrapper (class in src.common.types.pb_wrappers.subgraph_sampling_strategy) metadata() (src.common.graph_builder.pyg_graph_data.PygGraphData method) module common common.beam common.collections common.collections.frozen_dict common.collections.itertools common.collections.sorted_dict common.data common.data.export common.env_config common.logger common.metrics common.metrics.base_metrics common.metrics.decorators common.metrics.metrics_interface common.services common.services.dataproc common.services.gcp_cloud_logging common.services.kfp common.services.vertex_ai common.types common.types.resource_config common.types.uri common.types.uri.gcs_uri common.types.uri.http_uri common.types.uri.local_uri common.types.uri.uri common.types.uri.uri_factory common.types.wrappers common.types.wrappers.argo_workflow_manifest common.types.wrappers.kfp_api common.utils common.utils.compute common.utils.compute.random common.utils.compute.serialization common.utils.compute.serialization.coder common.utils.compute.serialization.serialize_np common.utils.compute.serialization.serialize_protos common.utils.compute.serialization.serialize_string common.utils.dataflow common.utils.func_tools common.utils.gcs common.utils.kfp common.utils.local_fs common.utils.os_utils common.utils.proto_utils common.utils.retry common.utils.test_utils common.utils.torch_training env env.dep_constants env.pipelines_config orchestration orchestration.kubeflow orchestration.kubeflow.kfp_orchestrator orchestration.kubeflow.kfp_pipeline orchestration.kubeflow.utils orchestration.kubeflow.utils.glt_backend orchestration.kubeflow.utils.log_metrics orchestration.kubeflow.utils.resource src src.applied_tasks src.applied_tasks.test_tasks src.applied_tasks.test_tasks.academic src.common src.common.constants src.common.constants.bq src.common.constants.components src.common.constants.gcs src.common.constants.graph_metadata src.common.constants.local_fs src.common.constants.metrics src.common.constants.resource_config src.common.constants.test_assets src.common.constants.time src.common.constants.training src.common.graph_builder src.common.graph_builder.abstract_graph_builder src.common.graph_builder.gbml_graph_protocol src.common.graph_builder.pyg_graph_builder src.common.graph_builder.pyg_graph_data src.common.modeling_task_specs src.common.modeling_task_specs.utils src.common.modeling_task_specs.utils.early_stop src.common.modeling_task_specs.utils.parameter_loading src.common.modeling_task_specs.utils.profiler_wrapper src.common.models src.common.models.dgl src.common.models.dgl.homogenous src.common.models.layers src.common.models.layers.count_min_sketch src.common.models.layers.decoder src.common.models.layers.feature_interaction src.common.models.layers.normalization src.common.models.pyg src.common.models.pyg.heterogeneous src.common.models.pyg.nn src.common.models.pyg.nn.conv src.common.models.pyg.nn.conv.edge_attr_gat_conv src.common.models.pyg.nn.conv.gin_conv src.common.models.pyg.nn.conv.hgt_conv src.common.models.pyg.nn.conv.simplehgn_conv src.common.models.pyg.nn.models src.common.models.pyg.nn.models.feature_interaction src.common.models.pyg.nn.models.jumping_knowledge src.common.models.pyg.utils src.common.models.utils src.common.models.utils.torch src.common.translators src.common.translators.gbml_protos_translator src.common.translators.model_eval_metrics_translator src.common.types src.common.types.dataflow_job_options src.common.types.dataset_split src.common.types.exception src.common.types.features src.common.types.graph_data src.common.types.model_eval_metrics src.common.types.pb_wrappers src.common.types.pb_wrappers.dataset_metadata src.common.types.pb_wrappers.flattened_graph_metadata src.common.types.pb_wrappers.gigl_resource_config src.common.types.pb_wrappers.graph_data_types src.common.types.pb_wrappers.graph_data_types_utils src.common.types.pb_wrappers.graph_metadata src.common.types.pb_wrappers.sampling_op src.common.types.pb_wrappers.subgraph_sampling_strategy src.common.types.pb_wrappers.task_metadata src.common.types.pb_wrappers.trained_model_metadata src.common.types.pb_wrappers.types src.common.types.task_metadata src.common.utils src.common.utils.bq src.common.utils.data src.common.utils.data.feature_serialization src.common.utils.dataflow src.common.utils.eval_metrics src.common.utils.file_loader src.common.utils.gbml_protos src.common.utils.metrics_service_provider src.common.utils.model src.common.utils.scheduler src.common.utils.spark_job_manager src.common.utils.time src.common.utils.timeout src.config_populator src.config_populator.config_populator src.data_preprocessor src.data_preprocessor.lib src.data_preprocessor.lib.enumerate src.data_preprocessor.lib.enumerate.queries src.data_preprocessor.lib.ingest src.data_preprocessor.lib.transform src.inference src.inference.v1 src.inference.v1.lib src.inference.v1.lib.inference_output_schema src.inference.v1.lib.queries src.inference.v1.lib.transforms src.inference.v2 src.post_process src.post_process.lib src.post_process.lib.base_post_processor src.post_process.utils src.post_process.utils.component_runtime src.post_process.utils.cosine_similarity src.split_generator src.subgraph_sampler src.subgraph_sampler.lib src.training src.training.v1 src.training.v1.lib src.training.v1.lib.data_loaders src.training.v1.lib.data_loaders.common src.training.v1.lib.eval_metrics src.training.v2 src.validation_check src.validation_check.libs src.validation_check.libs.resource_config_checks src.validation_check.libs.utils

modules() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) multi_get_tensor() (src.common.graph_builder.pyg_graph_data.PygGraphData method)

N

named_buffers() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) named_children() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) named_modules() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)

named_parameters() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) Node (class in src.common.types.graph_data) node_attrs() (src.common.graph_builder.pyg_graph_data.PygGraphData method) node_field (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema attribute) node_from_NodePb() (src.common.translators.gbml_protos_translator.GbmlProtosTranslator static method) node_items() (src.common.graph_builder.pyg_graph_data.PygGraphData method) node_stores (src.common.graph_builder.pyg_graph_data.PygGraphData property) node_type_subgraph() (src.common.graph_builder.pyg_graph_data.PygGraphData method) node_types (src.common.graph_builder.pyg_graph_data.PygGraphData property) NodePbWrapper (class in src.common.types.pb_wrappers.graph_data_types) NodePbWrapperCoder (class in common.utils.compute.serialization.serialize_protos) NopMetricsPublisher (class in common.metrics.base_metrics) num_edge_features (src.common.graph_builder.pyg_graph_data.PygGraphData property) num_edges (src.common.graph_builder.pyg_graph_data.PygGraphData property) num_features (src.common.graph_builder.pyg_graph_data.PygGraphData property) num_lines_in_file() (in module common.utils.os_utils) num_node_features (src.common.graph_builder.pyg_graph_data.PygGraphData property) num_nodes (src.common.graph_builder.pyg_graph_data.PygGraphData property) NumpyCoder (class in common.utils.compute.serialization.serialize_np)

O

OpsMetricPublisher (class in common.metrics.metrics_interface) orchestration module orchestration.kubeflow module orchestration.kubeflow.kfp_orchestrator module orchestration.kubeflow.kfp_pipeline module orchestration.kubeflow.utils module

orchestration.kubeflow.utils.glt_backend module orchestration.kubeflow.utils.log_metrics module orchestration.kubeflow.utils.resource module output_metadata (src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper property) (src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper property) output_metadata_type (src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper property) (src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper property)

P

parameters() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) parent_sampling_op_pb_wrappers (src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper property) parse_args() (in module common.utils.test_utils) parse_formatted_datetime() (in module src.common.utils.time) perm_assets_bucket_path (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) pin_memory() (src.common.graph_builder.pyg_graph_data.PygGraphData method)

pop() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) popitem() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) preprocessor_config (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) process() (common.logger.Logger method) profileit() (in module common.metrics.decorators) project (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) propagate() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) ProtoUtils (class in common.utils.proto_utils) put_edge_index() (src.common.graph_builder.pyg_graph_data.PygGraphData method) put_tensor() (src.common.graph_builder.pyg_graph_data.PygGraphData method) PygGraphBuilder (class in src.common.graph_builder.pyg_graph_builder) PygGraphData (class in src.common.graph_builder.pyg_graph_data)

R

RandomUniformPbWrapper (class in src.common.types.pb_wrappers.sampling_op) RandomWeightedPbWrapper (class in src.common.types.pb_wrappers.sampling_op) record_stream() (src.common.graph_builder.pyg_graph_data.PygGraphData method) region (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) register_aggregate_forward_hook() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_aggregate_forward_pre_hook() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_backward_hook() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_buffer() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_edge_types() (src.common.graph_builder.abstract_graph_builder.GraphBuilder method) (src.common.graph_builder.pyg_graph_builder.PygGraphBuilder method) register_edge_update_forward_hook() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_edge_update_forward_pre_hook() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_forward_hook() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_forward_pre_hook() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_full_backward_hook() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_full_backward_pre_hook() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_load_state_dict_post_hook() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)

register_message_and_aggregate_forward_hook() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_message_and_aggregate_forward_pre_hook() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_message_forward_hook() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_message_forward_pre_hook() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_module() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_parameter() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_propagate_forward_hook() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_propagate_forward_pre_hook() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_state_dict_pre_hook() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) register_structured_urn() (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder static method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder static method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder static method) register_urn() (common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder class method) (common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder class method) (common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder class method) relation (src.common.types.graph_data.EdgeType attribute) remove_edge_index() (src.common.graph_builder.pyg_graph_data.PygGraphData method) remove_file_if_exist() (in module common.utils.local_fs) remove_file_or_folder_if_exist() (in module common.utils.local_fs) remove_folder_if_exist() (in module common.utils.local_fs) remove_line_from_file() (in module common.utils.local_fs) remove_tensor() (src.common.graph_builder.pyg_graph_data.PygGraphData method) rename() (src.common.graph_builder.pyg_graph_data.PygGraphData method) requires_grad_() (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) reset() (src.common.graph_builder.abstract_graph_builder.GraphBuilder method) (src.common.graph_builder.pyg_graph_builder.PygGraphBuilder method) reset_parameters() (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) RetriesFailedException retry() (in module common.utils.retry) root_node_type_to_message_passing_path_pb_wrapper (src.common.types.pb_wrappers.subgraph_sampling_strategy.SubgraphSamplingStrategyPbWrapper property) run() (common.services.vertex_ai.VertexAIService method) (orchestration.kubeflow.kfp_orchestrator.KfpOrchestrator method) run_command_and_stream_stdout() (in module common.utils.os_utils) run_pipeline() (common.services.kfp.KFPService method) run_tests() (in module common.utils.test_utils)

S

SamplingDirectionEnumType (class in src.common.types.pb_wrappers.sampling_op) SamplingOpPbWrapper (class in src.common.types.pb_wrappers.sampling_op) save_scripted_model() (in module src.common.utils.model) save_state_dict() (in module src.common.utils.model) schema (src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema attribute) service_account_email (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) set_extra_state() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) set_value_dict() (src.common.graph_builder.pyg_graph_data.PygGraphData method) setdefault() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) setLevel() (common.logger.Logger method) share_memory() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) share_memory_() (src.common.graph_builder.pyg_graph_data.PygGraphData method) should_distribute() (in module common.utils.torch_training) SimpleHGN (class in src.common.models.pyg.heterogeneous) SimpleHGNConv (class in src.common.models.pyg.nn.conv.simplehgn_conv) size() (src.common.graph_builder.pyg_graph_data.PygGraphData method) snapshot() (src.common.graph_builder.pyg_graph_data.PygGraphData method) sort() (src.common.graph_builder.pyg_graph_data.PygGraphData method) sort_by_time() (src.common.graph_builder.pyg_graph_data.PygGraphData method) SortedDict (class in common.collections.sorted_dict) SparkJobManager (class in src.common.utils.spark_job_manager) split_generator_config (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) src module src.applied_tasks module src.applied_tasks.test_tasks module src.applied_tasks.test_tasks.academic module src.common module src.common.constants module src.common.constants.bq module src.common.constants.components module src.common.constants.gcs module src.common.constants.graph_metadata module src.common.constants.local_fs module src.common.constants.metrics module src.common.constants.resource_config module src.common.constants.test_assets module src.common.constants.time module src.common.constants.training module src.common.graph_builder module src.common.graph_builder.abstract_graph_builder module src.common.graph_builder.gbml_graph_protocol module src.common.graph_builder.pyg_graph_builder module src.common.graph_builder.pyg_graph_data module src.common.modeling_task_specs module src.common.modeling_task_specs.utils module src.common.modeling_task_specs.utils.early_stop module src.common.modeling_task_specs.utils.parameter_loading module src.common.modeling_task_specs.utils.profiler_wrapper module src.common.models module src.common.models.dgl module src.common.models.dgl.homogenous module src.common.models.layers module src.common.models.layers.count_min_sketch module src.common.models.layers.decoder module src.common.models.layers.feature_interaction module src.common.models.layers.normalization module src.common.models.pyg module src.common.models.pyg.heterogeneous module src.common.models.pyg.nn module src.common.models.pyg.nn.conv module src.common.models.pyg.nn.conv.edge_attr_gat_conv module src.common.models.pyg.nn.conv.gin_conv module src.common.models.pyg.nn.conv.hgt_conv module src.common.models.pyg.nn.conv.simplehgn_conv module src.common.models.pyg.nn.models module src.common.models.pyg.nn.models.feature_interaction module src.common.models.pyg.nn.models.jumping_knowledge module src.common.models.pyg.utils module src.common.models.utils module src.common.models.utils.torch module src.common.translators module src.common.translators.gbml_protos_translator module src.common.translators.model_eval_metrics_translator module src.common.types module src.common.types.dataflow_job_options module src.common.types.dataset_split module src.common.types.exception module

src.common.types.features module src.common.types.graph_data module src.common.types.model_eval_metrics module src.common.types.pb_wrappers module src.common.types.pb_wrappers.dataset_metadata module src.common.types.pb_wrappers.flattened_graph_metadata module src.common.types.pb_wrappers.gigl_resource_config module src.common.types.pb_wrappers.graph_data_types module src.common.types.pb_wrappers.graph_data_types_utils module src.common.types.pb_wrappers.graph_metadata module src.common.types.pb_wrappers.sampling_op module src.common.types.pb_wrappers.subgraph_sampling_strategy module src.common.types.pb_wrappers.task_metadata module src.common.types.pb_wrappers.trained_model_metadata module src.common.types.pb_wrappers.types module src.common.types.task_metadata module src.common.utils module src.common.utils.bq module src.common.utils.data module src.common.utils.data.feature_serialization module src.common.utils.dataflow module src.common.utils.eval_metrics module src.common.utils.file_loader module src.common.utils.gbml_protos module src.common.utils.metrics_service_provider module src.common.utils.model module src.common.utils.scheduler module src.common.utils.spark_job_manager module src.common.utils.time module src.common.utils.timeout module src.config_populator module src.config_populator.config_populator module src.data_preprocessor module src.data_preprocessor.lib module src.data_preprocessor.lib.enumerate module src.data_preprocessor.lib.enumerate.queries module src.data_preprocessor.lib.ingest module src.data_preprocessor.lib.transform module src.inference module src.inference.v1 module src.inference.v1.lib module src.inference.v1.lib.inference_output_schema module src.inference.v1.lib.queries module src.inference.v1.lib.transforms module src.inference.v2 module src.post_process module src.post_process.lib module src.post_process.lib.base_post_processor module src.post_process.utils module src.post_process.utils.component_runtime module src.post_process.utils.cosine_similarity module src.split_generator module src.subgraph_sampler module src.subgraph_sampler.lib module src.training module src.training.v1 module src.training.v1.lib module src.training.v1.lib.data_loaders module src.training.v1.lib.data_loaders.common module src.training.v1.lib.eval_metrics module src.training.v2 module src.validation_check module src.validation_check.libs module src.validation_check.libs.resource_config_checks module src.validation_check.libs.utils module src_node_type (src.common.types.graph_data.EdgeType attribute) state_dict() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) stores (src.common.graph_builder.pyg_graph_data.PygGraphData property) StringCoder (class in common.utils.compute.serialization.serialize_string) subgraph() (src.common.graph_builder.pyg_graph_data.PygGraphData method) subgraph_node_to_global_node_mapping (src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol property) (src.common.graph_builder.pyg_graph_data.PygGraphData property) subgraph_sampler_config (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) SubgraphSamplingStrategyPbWrapper (class in src.common.types.pb_wrappers.subgraph_sampling_strategy) SubgraphSamplingValidationError SubgraphSamplingValidationErrorType (class in src.common.types.exception) submit_and_wait_scala_spark_job() (common.services.dataproc.DataprocService method)

T

task_metadata (src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper property) task_metadata_type (src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper property) TaskMetadataPbWrapper (class in src.common.types.pb_wrappers.task_metadata) TaskMetadataType (class in src.common.types.task_metadata) temp_assets_bq_dataset_name (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) temp_assets_bucket_path (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) temp_assets_regional_bucket_path (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) test_file_pattern (common.utils.test_utils.TestArgs attribute) TestArgs (class in common.utils.test_utils) TimedOutException timeout() (in module src.common.utils.timeout) TimerRecordGranularity (class in common.metrics.decorators) to() (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) to_dict() (src.common.graph_builder.pyg_graph_data.PygGraphData method) to_empty() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)

to_hetero_data() (src.common.graph_builder.pyg_graph_data.PygGraphData method) to_hetero_feat() (in module src.common.models.utils.torch) to_homogeneous() (src.common.graph_builder.pyg_graph_data.PygGraphData method) to_namedtuple() (src.common.graph_builder.pyg_graph_data.PygGraphData method) TopKPbWrapper (class in src.common.types.pb_wrappers.sampling_op) TorchProfiler (class in src.common.modeling_task_specs.utils.profiler_wrapper) total() (src.common.models.layers.count_min_sketch.CountMinSketch method) train() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) TrainedModelMetadataPbWrapper (class in src.common.types.pb_wrappers.trained_model_metadata) trainer_config (src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) training_sample_type (src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper property) (src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper property) TrainingSamplesSchemaProtoUtils (class in src.common.utils.gbml_protos) type() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)

U

up_to() (src.common.graph_builder.pyg_graph_data.PygGraphData method) update() (common.utils.compute.serialization.serialize_np.EncodedNdArray method) (src.common.graph_builder.pyg_graph_data.PygGraphData method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) update_bq_dataset_retention() (src.common.utils.bq.BqUtils method) update_bq_table_retention() (src.common.utils.bq.BqUtils method)

update_bq_table_with_test_assets_and_version() (in module src.common.constants.test_assets) update_gcs_uri_with_test_assets_and_version() (in module src.common.constants.test_assets) update_tensor() (src.common.graph_builder.pyg_graph_data.PygGraphData method) upload() (orchestration.kubeflow.kfp_orchestrator.KfpOrchestrator method) upload_files_to_gcs() (common.utils.gcs.GcsUtils method) upload_from_filelike() (common.utils.gcs.GcsUtils method) upload_pipeline_version() (common.services.kfp.KFPService method) Uri (class in common.types.uri.uri) UriFactory (class in common.types.uri.uri_factory) UserDefinedPbWrapper (class in src.common.types.pb_wrappers.sampling_op)

V

validate() (src.common.graph_builder.pyg_graph_data.PygGraphData method) validate_dags() (src.common.types.pb_wrappers.subgraph_sampling_strategy.SubgraphSamplingStrategyPbWrapper method) values() (common.collections.frozen_dict.FrozenDict method) (common.collections.sorted_dict.SortedDict method) (common.utils.compute.serialization.serialize_np.EncodedNdArray method)

VertexAiJobConfig (class in common.services.vertex_ai) VertexAIService (class in common.services.vertex_ai) view() (src.common.graph_builder.pyg_graph_data.PygGraphData method) view_as() (src.common.types.dataflow_job_options.CommonOptions method)

W

wait_for_completion() (orchestration.kubeflow.kfp_orchestrator.KfpOrchestrator method)

wait_for_run_completion() (common.services.kfp.KFPService method) warning() (common.logger.Logger method)

X

xpu() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)

Z

zero_grad() (src.common.models.layers.decoder.LinkPredictionDecoder method) (src.common.models.layers.feature_interaction.DCNCross method) (src.common.models.layers.feature_interaction.DCNv2 method) (src.common.models.pyg.heterogeneous.HGT method) (src.common.models.pyg.heterogeneous.SimpleHGN method) (src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method)