Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | Y A a_etype (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) a_l (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) a_r (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) absolute() (gigl.common.LocalUri method) (gigl.common.types.uri.local_uri.LocalUri method) acc (gigl.src.common.types.model_eval_metrics.EvalMetricType attribute) accelerator_count (gigl.common.services.vertex_ai.VertexAiJobConfig attribute) accelerator_type (gigl.common.services.vertex_ai.VertexAiJobConfig attribute) ACT_MAP (in module gigl.src.common.constants.training) Action (class in gigl.orchestration.kubeflow.runner) activation (gigl.src.common.models.pyg.heterogeneous.SimpleHGN attribute) (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) activation_after_last_conv (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) activation_before_norm (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) active (gigl.src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler attribute) add() (gigl.src.common.models.layers.count_min_sketch.CountMinSketch method) add_bucket_lifecycle_rule_with_prefix() (gigl.common.utils.gcs.GcsUtils method) add_child_sampling_op_pb_wrapper() (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) add_confusion_matrix() (gigl.common.utils.kfp.KfpOutputViewers method) add_count() (gigl.common.metrics.base_metrics.NopMetricsPublisher method) (gigl.common.metrics.metrics_interface.OpsMetricPublisher method) add_edge() (gigl.src.common.graph_builder.abstract_graph_builder.GraphBuilder method) add_embedding() (gigl.common.data.export.EmbeddingExporter method) add_field() (gigl.src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchemaBuilder method) add_gauge() (gigl.common.metrics.base_metrics.NopMetricsPublisher method) (gigl.common.metrics.metrics_interface.OpsMetricPublisher method) add_graph_data() (gigl.src.common.graph_builder.abstract_graph_builder.GraphBuilder method) add_level() (gigl.common.metrics.base_metrics.NopMetricsPublisher method) (gigl.common.metrics.metrics_interface.OpsMetricPublisher method) add_metric() (gigl.src.common.types.model_eval_metrics.EvalMetricsCollection method) add_metrics() (gigl.src.common.types.model_eval_metrics.EvalMetricsCollection method) add_node() (gigl.src.common.graph_builder.abstract_graph_builder.GraphBuilder method) add_parent_sampling_op_pb_wrapper() (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) add_task() (gigl.src.common.models.layers.task.NodeAnchorBasedLinkPredictionTasks method) add_task_resource_requirements() (in module gigl.orchestration.kubeflow.utils.resource) add_timer() (gigl.common.metrics.base_metrics.NopMetricsPublisher method) (gigl.common.metrics.metrics_interface.OpsMetricPublisher method) add_torch_long_tensor() (gigl.src.common.models.layers.count_min_sketch.CountMinSketch method) additional_job_args (gigl.common.types.resource_config.CommonPipelineComponentConfigs attribute) AligmentLoss (class in gigl.src.common.models.layers.loss) alignment_loss (gigl.src.common.models.layers.task.DirectAU attribute) alpha (gigl.src.common.models.layers.loss.AligmentLoss attribute) (gigl.src.common.models.layers.loss.FeatureReconstructionLoss attribute) AnalyzeAndBuildTFTransformFn (class in gigl.src.data_preprocessor.lib.transform.utils) api_run (gigl.common.types.wrappers.kfp_api.ApiRunDetailWrapper property) ApiRunDetailWrapper (class in gigl.common.types.wrappers.kfp_api) append_line_to_file() (in module gigl.common.utils.local_fs) applied_task_identifier (gigl.orchestration.local.runner.PipelineConfig attribute) (gigl.src.config_populator.config_populator.ConfigPopulator property) (gigl.src.data_preprocessor.data_preprocessor.DataPreprocessor property) AppliedTaskIdentifier (in module gigl.src.common.types) are_disjoint() (gigl.src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol static method) are_same_graph() (gigl.src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol static method) args (gigl.common.services.vertex_ai.VertexAiJobConfig attribute) ArgumentReader (class in gigl.src.common.modeling_task_specs.utils.parameter_loading) assert_asset_exists() (in module gigl.src.validation_check.libs.frozen_config_path_checks) assert_component_runtimes_match_expected_parameters() (in module gigl.src.post_process.utils.component_runtime) assert_cosine_similarity_stats() (in module gigl.src.post_process.utils.cosine_similarity) assert_preprocessed_metadata_exists() (in module gigl.src.validation_check.libs.frozen_config_path_checks) assert_proto_field_value_is_truthy() (in module gigl.src.validation_check.libs.utils) assert_proto_has_field() (in module gigl.src.validation_check.libs.utils) assert_split_generator_output_exists() (in module gigl.src.validation_check.libs.frozen_config_path_checks) assert_subgraph_sampler_output_exists() (in module gigl.src.validation_check.libs.frozen_config_path_checks) assert_trained_model_exists() (in module gigl.src.validation_check.libs.frozen_config_path_checks) assigner_args (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig property) ASSIGNER_ARGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig attribute) assigner_cls_path (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig attribute) ASSIGNER_CLS_PATH_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig attribute) AVRO_SCHEMA (in module gigl.common.data.export) B BAD_INPUT_OP_NAME (gigl.src.common.types.exception.SubgraphSamplingValidationErrorType attribute) BaseInferenceBlueprint (class in gigl.src.inference.v1.lib.base_inference_blueprint) BaseInferencer (class in gigl.src.inference.v1.lib.base_inferencer) BaseIngestion (class in gigl.src.subgraph_sampler.lib.ingestion_protocol) BaseModelOperationsProtocol (class in gigl.src.common.types.model) BasePostProcessor (class in gigl.src.post_process.lib.base_post_processor) BaseTrainer (class in gigl.src.training.v1.lib.base_trainer) BasicHomogeneousGNN (class in gigl.src.common.models.pyg.homogeneous) batch() (in module gigl.common.collections.itertools) batch_combined_scores (gigl.src.common.models.layers.loss.ModelResultType attribute) (gigl.src.common.types.task_inputs.NodeAnchorBasedLinkPredictionTaskInputs attribute) batch_embeddings (gigl.src.common.models.layers.loss.ModelResultType attribute) (gigl.src.common.types.task_inputs.NodeAnchorBasedLinkPredictionTaskInputs attribute) batch_generator_fn (gigl.src.inference.v1.lib.transforms.batch_generator.BatchProcessorDoFn attribute) batch_scores (gigl.src.common.models.layers.loss.ModelResultType attribute) (gigl.src.common.types.task_inputs.NodeAnchorBasedLinkPredictionTaskInputs attribute) batch_size (gigl.common.data.dataloaders.TFDatasetOptions attribute) (gigl.src.training.v1.lib.data_loaders.common.DataloaderConfig attribute) BatchCombinedScores (class in gigl.src.common.types.task_inputs) BatchEmbeddings (class in gigl.src.common.types.task_inputs) batchnorm (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) BatchProcessorDoFn (class in gigl.src.inference.v1.lib.transforms.batch_generator) BatchScores (class in gigl.src.common.types.task_inputs) BatchType (in module gigl.src.inference.v1.lib.base_inference_blueprint) beam_record_tfxio (gigl.src.data_preprocessor.lib.transform.utils.IngestRawFeatures attribute) best_criterion (gigl.src.common.modeling_task_specs.utils.early_stop.EarlyStopper property) best_model_state_dict (gigl.src.common.modeling_task_specs.utils.early_stop.EarlyStopper property) BetterWriteToTFRecord (class in gigl.common.beam.better_tfrecordio) BGRL (class in gigl.src.common.models.layers.task) BGRLLoss (class in gigl.src.common.models.layers.loss) BIGQUERY_SCHEMA (in module gigl.common.data.export) BigqueryEdgeDataReference (class in gigl.src.data_preprocessor.lib.ingest.bigquery) BigqueryNodeDataReference (class in gigl.src.data_preprocessor.lib.ingest.bigquery) bq_inferencer_output_paths (gigl.src.inference.v1.gnn_inferencer.InferencerOutputPaths attribute) bq_unique_node_ids_enumerated_table_name (gigl.src.data_preprocessor.lib.enumerate.utils.EnumeratorNodeTypeMetadata attribute) bq_utils (gigl.src.inference.v1.gnn_inferencer.InferencerV1 property) BQGraphValidator (class in gigl.analytics.graph_validation) (class in gigl.analytics.graph_validation.bq_graph_validator) BqUtils (class in gigl.src.common.utils.bq) bucket (gigl.common.GcsUri property) (gigl.common.types.uri.gcs_uri.GcsUri property) build() (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset method) (gigl.distributed.DistLinkPredictionDataset method) (gigl.src.common.graph_builder.abstract_graph_builder.GraphBuilder method) (gigl.src.common.graph_builder.pyg_graph_builder.PygGraphBuilder method) (gigl.src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchemaBuilder method) build_and_write_node_anchor_link_prediction_subgraph_samples_from_mocked_dataset_info() (in module gigl.src.mocking.lib.mock_input_for_split_generator) build_and_write_supervised_node_classification_subgraph_samples_from_mocked_dataset_info() (in module gigl.src.mocking.lib.mock_input_for_split_generator) build_dataset() (in module gigl.distributed) (in module gigl.distributed.dataset_factory) build_dataset_from_task_config_uri() (in module gigl.distributed) (in module gigl.distributed.dataset_factory) build_dry_subgraph() (gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper class method) build_ingestion_feature_spec_fn() (in module gigl.src.data_preprocessor.lib.data_preprocessor_config) build_k_hop_subgraphs_from_pyg_heterodata() (in module gigl.src.mocking.lib.pyg_to_training_samples) build_node_anchor_link_prediction_samples_from_pyg_heterodata() (in module gigl.src.mocking.lib.pyg_to_training_samples) build_NodeAnchorBasedLinkPredictionSamplePb() (gigl.src.common.utils.gbml_protos.TrainingSamplesSchemaProtoUtils static method) build_passthrough_transform_preprocessing_fn() (in module gigl.src.data_preprocessor.lib.data_preprocessor_config) build_pyg_heterodata_from_mocked_dataset_info() (in module gigl.src.mocking.lib.pyg_to_training_samples) build_RootedNodeNeighborhoodPb() (gigl.src.common.utils.gbml_protos.TrainingSamplesSchemaProtoUtils static method) build_supervised_node_classification_samples_from_pyg_heterodata() (in module gigl.src.mocking.lib.pyg_to_training_samples) build_SupervisedLinkBasedTaskSamplePb() (gigl.src.common.utils.gbml_protos.TrainingSamplesSchemaProtoUtils static method) build_SupervisedNodeClassificationSamplePb() (gigl.src.common.utils.gbml_protos.TrainingSamplesSchemaProtoUtils static method) C cache_mappings() (in module gigl.src.inference.v1.lib.transforms.utils) calculate_batch_retrieval_loss() (gigl.src.common.models.layers.loss.RetrievalLoss method) calculate_cosine_sim_between_embedding_tables() (in module gigl.src.post_process.utils.cosine_similarity) calculate_cosine_similarity_stats() (in module gigl.src.post_process.utils.cosine_similarity) calculate_in_batch_candidate_sampling_probability() (in module gigl.src.common.models.layers.count_min_sketch) calculate_losses() (gigl.src.common.models.layers.task.NodeAnchorBasedLinkPredictionTasks method) cast_graph_for_training() (in module gigl.src.training.v1.lib.data_loaders.utils) change_working_dir_to_gigl_root() (in module gigl.common.utils.jupyter_magics) check_columns_exist_in_table() (gigl.src.common.utils.bq.BqUtils method) check_glt_backend_eligibility_component() (in module gigl.orchestration.kubeflow.utils.glt_backend) check_if_dag_contains_cycles() (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) check_if_data_preprocessor_config_cls_valid() (in module gigl.src.validation_check.libs.template_config_checks) check_if_graph_metadata_valid() (in module gigl.src.validation_check.libs.template_config_checks) check_if_inferencer_cls_valid() (in module gigl.src.validation_check.libs.template_config_checks) check_if_inferencer_resource_config_valid() (in module gigl.src.validation_check.libs.resource_config_checks) check_if_kfp_pipeline_job_name_valid() (in module gigl.src.validation_check.libs.name_checks) (in module gigl.src.validation_check.libs.template_config_checks) check_if_post_processor_cls_valid() (in module gigl.src.validation_check.libs.template_config_checks) check_if_preprocessed_metadata_valid() (in module gigl.src.validation_check.libs.template_config_checks) check_if_preprocessor_resource_config_valid() (in module gigl.src.validation_check.libs.resource_config_checks) check_if_runtime_args_all_str() (in module gigl.src.validation_check.libs.template_config_checks) check_if_shared_resource_config_valid() (in module gigl.src.validation_check.libs.resource_config_checks) check_if_split_generator_config_valid() (in module gigl.src.validation_check.libs.template_config_checks) check_if_split_generator_resource_config_valid() (in module gigl.src.validation_check.libs.resource_config_checks) check_if_subgraph_sampler_config_valid() (in module gigl.src.validation_check.libs.template_config_checks) check_if_subgraph_sampler_resource_config_valid() (in module gigl.src.validation_check.libs.resource_config_checks) check_if_task_metadata_valid() (in module gigl.src.validation_check.libs.template_config_checks) check_if_trainer_cls_valid() (in module gigl.src.validation_check.libs.template_config_checks) check_if_trainer_resource_config_valid() (in module gigl.src.validation_check.libs.resource_config_checks) check_pipeline_has_valid_start_and_stop_flags() (in module gigl.src.validation_check.libs.template_config_checks) check_sampling_op_edge_type_validity() (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper method) child_sampling_op_pb_wrappers (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper property) clean_up() (gigl.src.subgraph_sampler.lib.ingestion_protocol.BaseIngestion method) cleanup_dataloaders() (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.NodeAnchorBasedLinkPredictionDatasetDataloaders method) (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.SupervisedNodeClassificationDatasetDataloaders method) ClearField() (snapchat.research.gbml.dataset_metadata_pb2.DatasetMetadata method) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset method) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.TestNodeTypeToRandomNegativeDataUriEntry method) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.TrainNodeTypeToRandomNegativeDataUriEntry method) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.ValNodeTypeToRandomNegativeDataUriEntry method) (snapchat.research.gbml.dataset_metadata_pb2.SupervisedLinkBasedTaskSplitDataset method) (snapchat.research.gbml.dataset_metadata_pb2.SupervisedNodeClassificationDataset method) (snapchat.research.gbml.flattened_graph_metadata_pb2.FlattenedGraphMetadata method) (snapchat.research.gbml.flattened_graph_metadata_pb2.NodeAnchorBasedLinkPredictionOutput method) (snapchat.research.gbml.flattened_graph_metadata_pb2.NodeAnchorBasedLinkPredictionOutput.NodeTypeToRandomNegativeTfrecordUriPrefixEntry method) (snapchat.research.gbml.flattened_graph_metadata_pb2.SupervisedLinkBasedTaskOutput method) (snapchat.research.gbml.flattened_graph_metadata_pb2.SupervisedNodeClassificationOutput method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.DataPreprocessorConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.FeatureFlagsEntry method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDbArgsEntry method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDBServiceConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig.InferencerArgsEntry method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.MetricsConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.MetricsConfig.MetricsArgsEntry method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.PostProcessorConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig.ProfilerArgsEntry method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig.SharedArgsEntry method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig.TrainerArgsEntry method) (snapchat.research.gbml.gigl_resource_config_pb2.DataflowResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.DataPreprocessorConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.DistributedTrainerConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.InferencerResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.KFPResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.KFPTrainerConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.LocalResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.LocalTrainerConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.ResourceLabelsEntry method) (snapchat.research.gbml.gigl_resource_config_pb2.SparkResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.TrainerResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiTrainerConfig method) (snapchat.research.gbml.graph_schema_pb2.Edge method) (snapchat.research.gbml.graph_schema_pb2.EdgeType method) (snapchat.research.gbml.graph_schema_pb2.Graph method) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata method) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedEdgeTypeMapEntry method) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedNodeTypeMapEntry method) (snapchat.research.gbml.graph_schema_pb2.Node method) (snapchat.research.gbml.inference_metadata_pb2.InferenceMetadata method) (snapchat.research.gbml.inference_metadata_pb2.InferenceMetadata.NodeTypeToInferencerOutputInfoMapEntry method) (snapchat.research.gbml.inference_metadata_pb2.InferenceOutput method) (snapchat.research.gbml.postprocessed_metadata_pb2.PostProcessedMetadata method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedEdgeTypeToPreprocessedMetadataEntry method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedNodeTypeToPreprocessedMetadataEntry method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.GlobalRandomUniformStrategy method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.MessagePassingPath method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.MessagePassingPathStrategy method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.RandomUniform method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.RandomWeighted method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SubgraphSamplingStrategy method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.TopK method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.UserDefined method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.UserDefined.ParamsEntry method) (snapchat.research.gbml.trained_model_metadata_pb2.TrainedModelMetadata method) (snapchat.research.gbml.training_samples_schema_pb2.Label method) (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample method) (snapchat.research.gbml.training_samples_schema_pb2.RootedNodeNeighborhood method) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedLinkBasedTaskSample method) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedNodeClassificationSample method) clip_grad_norm (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec attribute) close_upload_delete_and_push_to_gcs() (gigl.common.utils.gcs.GcsUtils method) CLOUD_LOGGING_DATETIME_FORMAT (in module gigl.common.services.gcp_cloud_logging) cls_path (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig attribute) CLS_PATH_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig attribute) cluster_client (gigl.common.services.dataproc.DataprocService attribute) cluster_name (gigl.src.common.utils.spark_job_manager.DataprocClusterInitData attribute) coder (gigl.src.common.utils.data.feature_serialization.FeatureSerializationUtils attribute) CoderProtocol (class in gigl.common.utils.compute.serialization.coder) collate_pyg_node_anchor_based_link_prediction_minibatch() (gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader.NodeAnchorBasedLinkPredictionBatch static method) collate_pyg_node_classification_minibatch() (gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader.SupervisedNodeClassificationBatch static method) collate_pyg_rooted_node_neighborhood_minibatch() (gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader.RootedNodeNeighborhoodBatch static method) combination_mode (gigl.src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction attribute) CombinationMode (class in gigl.src.common.models.pyg.nn.models.feature_interaction) CombinedIterableDatasets (class in gigl.src.training.v1.lib.data_loaders.tf_records_iterable_dataset) command (gigl.common.services.vertex_ai.VertexAiJobConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig attribute) COMMAND_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig attribute) common_compute_config (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig property) COMMON_COMPUTE_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig attribute) CommonOptions (class in gigl.src.common.types.dataflow_job_options) CommonPipelineComponentConfigs (class in gigl.common.types.resource_config) COMPILE (gigl.orchestration.kubeflow.runner.Action attribute) compile() (gigl.orchestration.kubeflow.kfp_orchestrator.KfpOrchestrator class method) Component (class in snapchat.research.gbml.gigl_resource_config_pb2) Component_Config_Populator (in module snapchat.research.gbml.gigl_resource_config_pb2) COMPONENT_CONFIG_POPULATOR_PREFIX (in module gigl.src.common.constants.metrics) Component_Config_Validator (in module snapchat.research.gbml.gigl_resource_config_pb2) Component_Data_Preprocessor (in module snapchat.research.gbml.gigl_resource_config_pb2) Component_Inferencer (in module snapchat.research.gbml.gigl_resource_config_pb2) COMPONENT_INFERENCER_PREFIX (in module gigl.src.common.constants.metrics) COMPONENT_POSTPROCESSOR_PREFIX (in module gigl.src.common.constants.metrics) COMPONENT_PREPROCESSOR_PREFIX (in module gigl.src.common.constants.metrics) Component_Split_Generator (in module snapchat.research.gbml.gigl_resource_config_pb2) COMPONENT_SPLIT_GENERATOR_PREFIX (in module gigl.src.common.constants.metrics) Component_Subgraph_Sampler (in module snapchat.research.gbml.gigl_resource_config_pb2) COMPONENT_SUBGRAPH_SAMPLER_PREFIX (in module gigl.src.common.constants.metrics) COMPONENT_TO_SHORTENED_COST_LABEL_MAP (in module gigl.src.common.types.pb_wrappers.gigl_resource_config) Component_Trainer (in module snapchat.research.gbml.gigl_resource_config_pb2) COMPONENT_TRAINER_PREFIX (in module gigl.src.common.constants.metrics) Component_Unknown (in module snapchat.research.gbml.gigl_resource_config_pb2) compute_datasets_to_mock() (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite method) condensed_edge_type (gigl.src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper property) (snapchat.research.gbml.graph_schema_pb2.Edge attribute) CONDENSED_EDGE_TYPE_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.Edge attribute) condensed_edge_type_map (snapchat.research.gbml.graph_schema_pb2.GraphMetadata property) CONDENSED_EDGE_TYPE_MAP_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.GraphMetadata attribute) condensed_edge_type_to_condensed_node_types (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) condensed_edge_type_to_edge_type_map (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) condensed_edge_type_to_feature_dim_map (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) condensed_edge_type_to_feature_keys_map (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) condensed_edge_type_to_feature_schema_map (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) condensed_edge_type_to_hard_neg_edge_feature_dim_map (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) condensed_edge_type_to_hard_neg_edge_feature_keys_map (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) condensed_edge_type_to_hard_neg_edge_feature_schema_map (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) condensed_edge_type_to_pos_edge_feature_dim_map (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) condensed_edge_type_to_pos_edge_feature_keys_map (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) condensed_edge_type_to_pos_edge_feature_schema_map (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) condensed_edge_type_to_preprocessed_metadata (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata property) CONDENSED_EDGE_TYPE_TO_PREPROCESSED_METADATA_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata attribute) condensed_edge_type_to_supervision_edge_data (gigl.src.common.translators.training_samples_protos_translator.NodeAnchorBasedLinkPredictionSample attribute) condensed_edge_types (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) condensed_node_type (gigl.src.common.types.pb_wrappers.graph_data_types.NodePbWrapper property) (snapchat.research.gbml.graph_schema_pb2.Node attribute) CONDENSED_NODE_TYPE_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.Node attribute) condensed_node_type_map (snapchat.research.gbml.graph_schema_pb2.GraphMetadata property) CONDENSED_NODE_TYPE_MAP_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.GraphMetadata attribute) condensed_node_type_to_feature_dim_map (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) condensed_node_type_to_feature_keys_map (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) condensed_node_type_to_feature_schema_map (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) condensed_node_type_to_node_type_map (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) condensed_node_type_to_preprocessed_metadata (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata property) CONDENSED_NODE_TYPE_TO_PREPROCESSED_METADATA_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata attribute) condensed_node_type_to_root_node_indices_map (gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader.RootedNodeNeighborhoodBatch attribute) condensed_node_type_to_subgraph_id_to_global_node_id (gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader.NodeAnchorBasedLinkPredictionBatch attribute) (gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader.RootedNodeNeighborhoodBatch attribute) condensed_node_types (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) CondensedEdgeType (in module gigl.src.common.types.graph_data) CondensedNodeType (in module gigl.src.common.types.graph_data) config_check() (gigl.orchestration.local.runner.Runner static method) ConfigPopulator (class in gigl.src.config_populator.config_populator) (gigl.src.common.constants.components.GiGLComponents attribute) ConfigValidator (gigl.src.common.constants.components.GiGLComponents attribute) connect_worker_pool() (in module gigl.common.utils.vertex_ai_context) container_uri (gigl.common.services.vertex_ai.VertexAiJobConfig attribute) CONTAINS_INVALID_EDGE_IN_DAG (gigl.src.common.types.exception.SubgraphSamplingValidationErrorType attribute) conv1 (gigl.src.common.models.pyg.homogeneous.TwoLayerGCN attribute) conv2 (gigl.src.common.models.pyg.homogeneous.TwoLayerGCN attribute) conv_layers (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) convert_days_to_ms() (in module gigl.src.common.utils.time) convert_pb_to_serialized_graph_metadata() (in module gigl.distributed.utils.serialized_graph_metadata_translator) convs (gigl.src.common.models.pyg.heterogeneous.HGT attribute) (gigl.src.common.models.pyg.heterogeneous.SimpleHGN attribute) copy_files() (in module gigl.common.utils.local_fs) copy_gcs_path() (gigl.common.utils.gcs.GcsUtils method) CORA_NODE_ANCHOR_MOCKED_DATASET_INFO (in module gigl.src.mocking.mocking_assets.mocked_datasets_for_pipeline_tests) CORA_NODE_CLASSIFICATION_MOCKED_DATASET_INFO (in module gigl.src.mocking.mocking_assets.mocked_datasets_for_pipeline_tests) CORA_USER_DEFINED_NODE_ANCHOR_MOCKED_DATASET_INFO (in module gigl.src.mocking.mocking_assets.mocked_datasets_for_pipeline_tests) CoraFromGCS (class in gigl.src.mocking.lib.pyg_datasets_forks) COSINE_SIM_FIELD (in module gigl.src.post_process.utils.cosine_similarity) CosineDecayScheduler (class in gigl.src.common.utils.scheduler) count_assets() (gigl.src.common.utils.file_loader.FileLoader method) count_blobs_in_gcs_path() (gigl.common.utils.gcs.GcsUtils method) count_files_with_uri_prefix() (in module gigl.common.utils.local_fs) count_number_of_columns_in_bq_table() (gigl.src.common.utils.bq.BqUtils method) count_number_of_rows_in_bq_table() (gigl.src.common.utils.bq.BqUtils method) COUNT_SGS_USES_GRAPHDB (in module gigl.src.common.constants.metrics) CountMinSketch (class in gigl.src.common.models.layers.count_min_sketch) counts (gigl.common.metrics.base_metrics.NopMetricsPublisher attribute) cpu_container_image (gigl.common.types.resource_config.CommonPipelineComponentConfigs attribute) cpu_request (snapchat.research.gbml.gigl_resource_config_pb2.KFPResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.KFPTrainerConfig attribute) CPU_REQUEST_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.KFPResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.KFPTrainerConfig attribute) create_bq_dataset() (gigl.src.common.utils.bq.BqUtils method) create_cluster() (gigl.common.services.dataproc.DataprocService method) create_dataproc_cluster() (gigl.src.common.utils.spark_job_manager.SparkJobManager method) create_empty_file_if_none_exists() (in module gigl.common.utils.local_fs) create_file_symlinks() (in module gigl.common.utils.local_fs) create_or_empty_bq_table() (gigl.src.common.utils.bq.BqUtils method) create_uri() (gigl.common.types.uri.uri_factory.UriFactory static method) (gigl.common.UriFactory static method) created_at (gigl.common.types.wrappers.kfp_api.ApiRunDetailWrapper property) cuda_container_image (gigl.common.types.resource_config.CommonPipelineComponentConfigs attribute) current_datetime() (in module gigl.src.common.utils.time) current_formatted_datetime() (in module gigl.src.common.utils.time) custom_cpu_docker_uri (gigl.orchestration.local.runner.PipelineConfig attribute) custom_cuda_docker_uri (gigl.orchestration.local.runner.PipelineConfig attribute) custom_identifier (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) custom_worker_image_uri (gigl.src.data_preprocessor.data_preprocessor.DataPreprocessor property) D DAG_CONTAINS_CYCLE (gigl.src.common.types.exception.SubgraphSamplingValidationErrorType attribute) DASH_DATE_FMT (in module gigl.src.common.constants.time) data (gigl.common.utils.compute.serialization.serialize_np.EncodedNdArray attribute) data_preprocessor_args (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.DataPreprocessorConfig property) DATA_PREPROCESSOR_ARGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.DataPreprocessorConfig attribute) data_preprocessor_config (gigl.src.data_preprocessor.data_preprocessor.DataPreprocessor property) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig property) data_preprocessor_config_cls_path (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.DataPreprocessorConfig attribute) DATA_PREPROCESSOR_CONFIG_CLS_PATH_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.DataPreprocessorConfig attribute) DATA_PREPROCESSOR_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig attribute) data_reference (gigl.src.data_preprocessor.lib.transform.utils.IngestRawFeatures attribute) dataflow_console_uri (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) dataflow_container_image (gigl.common.types.resource_config.CommonPipelineComponentConfigs attribute) dataflow_docker_uri (gigl.orchestration.local.runner.PipelineConfig attribute) dataflow_inferencer_config (snapchat.research.gbml.gigl_resource_config_pb2.InferencerResourceConfig property) DATAFLOW_INFERENCER_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.InferencerResourceConfig attribute) dataflow_runner (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) DATAFLOW_RUNNER_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) DATAFLOW_SPLIT_GENERATOR_CONFIG (in module gigl.src.common.constants.resource_config) DataflowResourceConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) DataloaderConfig (class in gigl.src.training.v1.lib.data_loaders.common) Dataloaders (class in gigl.src.common.types.pb_wrappers.dataset_metadata_utils) dataloaders (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.NodeAnchorBasedLinkPredictionDatasetDataloaders attribute) (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.SupervisedNodeClassificationDatasetDataloaders attribute) DataloaderTypes (class in gigl.src.training.v1.lib.data_loaders.common) DataPreprocessor (class in gigl.src.data_preprocessor.data_preprocessor) (gigl.src.common.constants.components.GiGLComponents attribute) DataPreprocessorConfig (class in gigl.src.data_preprocessor.lib.data_preprocessor_config) (class in snapchat.research.gbml.gigl_resource_config_pb2) DataprocClusterInitData (class in gigl.src.common.utils.spark_job_manager) DataprocService (class in gigl.common.services.dataproc) DataReference (class in gigl.src.data_preprocessor.lib.ingest.reference) dataset_config (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig property) DATASET_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig attribute) dataset_metadata (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig property) DATASET_METADATA_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) dataset_metadata_pb (gigl.src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper attribute) dataset_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) DATASET_TO_TRAINING_SAMPLE_TYPE (in module gigl.src.common.types.pb_wrappers.dataset_metadata) dataset_type (gigl.src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper property) DatasetAssetMocker (class in gigl.src.mocking.dataset_asset_mocker) DatasetAssetMockingSuite (class in gigl.src.mocking.dataset_asset_mocking_suite) DatasetAssetMockingSuite.ToyGraphData (class in gigl.src.mocking.dataset_asset_mocking_suite) DatasetAssetMockingSuite.UserDefinedLabels (class in gigl.src.mocking.dataset_asset_mocking_suite) DatasetMetadata (class in snapchat.research.gbml.dataset_metadata_pb2) DatasetMetadataPb (in module gigl.src.common.types.pb_wrappers.types) DatasetMetadataPbWrapper (class in gigl.src.common.types.pb_wrappers.dataset_metadata) DatasetSplit (class in gigl.src.common.types.dataset_split) DBLP_GRAPH_NODE_ANCHOR_MOCKED_DATASET_INFO (in module gigl.src.mocking.mocking_assets.mocked_datasets_for_pipeline_tests) DBLPFromGCS (class in gigl.src.mocking.lib.pyg_datasets_forks) DCNCross (class in gigl.src.common.models.layers.feature_interaction) DCNv2 (class in gigl.src.common.models.layers.feature_interaction) dcnv2 (gigl.src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction attribute) debug_cluster_owner_alias (gigl.src.common.utils.spark_job_manager.DataprocClusterInitData attribute) decode() (gigl.common.beam.coders.PassthroughCoder method) (gigl.common.utils.compute.serialization.coder.CoderProtocol method) (gigl.common.utils.compute.serialization.serialize_np.NumpyCoder method) (gigl.common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (gigl.common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (gigl.common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) (gigl.common.utils.compute.serialization.serialize_string.StringCoder method) (gigl.src.common.models.pyg.link_prediction.LinkPredictionGNN method) decoder_channel_list (gigl.src.common.models.layers.decoder.LinkPredictionDecoder attribute) decoder_type (gigl.src.common.models.layers.decoder.LinkPredictionDecoder attribute) DecoderType (class in gigl.src.common.models.layers.decoder) DEFAULT_BATCH_SIZE (in module gigl.src.inference.v1.lib.utils) DEFAULT_CONDENSED_EDGE_TYPE (in module gigl.src.common.constants.graph_metadata) DEFAULT_CONDENSED_NODE_TYPE (in module gigl.src.common.constants.graph_metadata) DEFAULT_CUSTOM_JOB_TIMEOUT_S (in module gigl.common.services.vertex_ai) DEFAULT_DATE_FORMAT (in module gigl.src.common.constants.time) DEFAULT_DATETIME_FORMAT (in module gigl.src.common.constants.time) default_edge_type (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo property) DEFAULT_EMBEDDING_FIELD (in module gigl.src.inference.v1.lib.inference_output_schema) DEFAULT_EMBEDDINGS_TABLE_SCHEMA (in module gigl.src.inference.v1.lib.inference_output_schema) DEFAULT_ENUMERATED_NODE_ID_FIELD (in module gigl.src.data_preprocessor.lib.enumerate.queries) DEFAULT_HOMOGENEOUS_EDGE_TYPE (in module gigl.types.graph) DEFAULT_HOMOGENEOUS_NODE_TYPE (in module gigl.types.graph) DEFAULT_JOB_NAME (in module gigl.orchestration.kubeflow.runner) DEFAULT_KFP_COMPILED_PIPELINE_DEST_PATH (in module gigl.orchestration.kubeflow.kfp_orchestrator) DEFAULT_MASTER_DATA_BUILDING_PORT (in module gigl.distributed.constants) DEFAULT_MASTER_INFERENCE_PORT (in module gigl.distributed.constants) DEFAULT_MASTER_SAMPLING_PORT (in module gigl.distributed.constants) DEFAULT_NODE_ID_FIELD (in module gigl.src.inference.v1.lib.inference_output_schema) default_node_type (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo property) DEFAULT_NUM_CPU_THREADS (in module gigl.distributed.distributed_neighborloader) DEFAULT_NUM_GNN_HOPS (in module gigl.src.common.constants.training) DEFAULT_NUM_HOPS_FOR_DATASETS (in module gigl.src.mocking.lib.pyg_to_training_samples) DEFAULT_NUM_NEGATIVE_SAMPLES_PER_POS_EDGE (in module gigl.src.mocking.lib.pyg_to_training_samples) DEFAULT_NUM_NODES_PER_HOP (in module gigl.src.mocking.lib.pyg_to_training_samples) DEFAULT_ORIGINAL_NODE_ID_FIELD (in module gigl.src.data_preprocessor.lib.enumerate.queries) DEFAULT_PIPELINE_TIMEOUT_S (in module gigl.common.services.vertex_ai) DEFAULT_PIPELINE_VERSION_NAME (in module gigl.orchestration.kubeflow.kfp_orchestrator) DEFAULT_PREDICTION_FIELD (in module gigl.src.inference.v1.lib.inference_output_schema) DEFAULT_PREDICTIONS_TABLE_SCHEMA (in module gigl.src.inference.v1.lib.inference_output_schema) DEFAULT_PYG_NODE_ANCHOR_SPLIT_TRANSFORM (in module gigl.src.mocking.lib.pyg_to_training_samples) DEFAULT_START_AT (in module gigl.orchestration.kubeflow.runner) DEFAULT_START_AT_COMPONENT (in module gigl.orchestration.kubeflow.kfp_orchestrator) DEFAULT_TF_FLOAT_DTYPE (in module gigl.src.data_preprocessor.lib.types) DEFAULT_TF_INT_DTYPE (in module gigl.src.data_preprocessor.lib.types) DEFAULT_TF_STRING_DTYPE (in module gigl.src.data_preprocessor.lib.types) DEFAULT_VERTEX_AI_TIMEOUT_S (in module gigl.src.training.v2.glt_trainer) dehydrate() (gigl.src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper method) (gigl.src.common.types.pb_wrappers.graph_data_types.NodePbWrapper method) delete_and_create_local_path() (in module gigl.common.utils.local_fs) delete_bq_table_if_exist() (gigl.src.common.utils.bq.BqUtils method) delete_cluster() (gigl.common.services.dataproc.DataprocService method) (gigl.src.common.utils.spark_job_manager.SparkJobManager method) delete_files() (gigl.common.utils.gcs.GcsUtils method) (gigl.src.common.utils.file_loader.FileLoader method) delete_files_in_bucket_dir() (gigl.common.utils.gcs.GcsUtils method) delete_gcs_file_if_exist() (gigl.common.utils.gcs.GcsUtils method) delete_local_directory() (in module gigl.common.utils.local_fs) delete_matching_tables() (gigl.src.common.utils.bq.BqUtils method) DELIMITER (in module gigl.src.common.constants.metrics) DESCRIPTOR (in module snapchat.research.gbml.dataset_metadata_pb2) (in module snapchat.research.gbml.flattened_graph_metadata_pb2) (in module snapchat.research.gbml.gbml_config_pb2) (in module snapchat.research.gbml.gigl_resource_config_pb2) (in module snapchat.research.gbml.graph_schema_pb2) (in module snapchat.research.gbml.inference_metadata_pb2) (in module snapchat.research.gbml.postprocessed_metadata_pb2) (in module snapchat.research.gbml.preprocessed_metadata_pb2) (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) (in module snapchat.research.gbml.trained_model_metadata_pb2) (in module snapchat.research.gbml.training_samples_schema_pb2) (snapchat.research.gbml.dataset_metadata_pb2.DatasetMetadata attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.TestNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.TrainNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.ValNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.dataset_metadata_pb2.SupervisedLinkBasedTaskSplitDataset attribute) (snapchat.research.gbml.dataset_metadata_pb2.SupervisedNodeClassificationDataset attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.FlattenedGraphMetadata attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.NodeAnchorBasedLinkPredictionOutput attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.NodeAnchorBasedLinkPredictionOutput.NodeTypeToRandomNegativeTfrecordUriPrefixEntry attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.SupervisedLinkBasedTaskOutput attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.SupervisedNodeClassificationOutput attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.DataPreprocessorConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.FeatureFlagsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDbArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDBServiceConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig.InferencerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.MetricsConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.MetricsConfig.MetricsArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.PostProcessorConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig.ProfilerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig.SharedArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig.TrainerArgsEntry attribute) (snapchat.research.gbml.gigl_resource_config_pb2.DataflowResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.DataPreprocessorConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.DistributedTrainerConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.InferencerResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.KFPResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.KFPTrainerConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.LocalResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.LocalTrainerConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.ResourceLabelsEntry attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SparkResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.TrainerResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiTrainerConfig attribute) (snapchat.research.gbml.graph_schema_pb2.Edge attribute) (snapchat.research.gbml.graph_schema_pb2.EdgeType attribute) (snapchat.research.gbml.graph_schema_pb2.Graph attribute) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata attribute) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedEdgeTypeMapEntry attribute) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedNodeTypeMapEntry attribute) (snapchat.research.gbml.graph_schema_pb2.Node attribute) (snapchat.research.gbml.inference_metadata_pb2.InferenceMetadata attribute) (snapchat.research.gbml.inference_metadata_pb2.InferenceMetadata.NodeTypeToInferencerOutputInfoMapEntry attribute) (snapchat.research.gbml.inference_metadata_pb2.InferenceOutput attribute) (snapchat.research.gbml.postprocessed_metadata_pb2.PostProcessedMetadata attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedEdgeTypeToPreprocessedMetadataEntry attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedNodeTypeToPreprocessedMetadataEntry attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.GlobalRandomUniformStrategy attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.MessagePassingPath attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.MessagePassingPathStrategy attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.RandomUniform attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.RandomWeighted attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SubgraphSamplingStrategy attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.TopK attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.UserDefined attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.UserDefined.ParamsEntry attribute) (snapchat.research.gbml.trained_model_metadata_pb2.TrainedModelMetadata attribute) (snapchat.research.gbml.training_samples_schema_pb2.Label attribute) (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample attribute) (snapchat.research.gbml.training_samples_schema_pb2.RootedNodeNeighborhood attribute) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedLinkBasedTaskSample attribute) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedNodeClassificationSample attribute) deserialize_edge_features() (gigl.src.common.utils.data.feature_serialization.FeatureSerializationUtils class method) deserialize_node_features() (gigl.src.common.utils.data.feature_serialization.FeatureSerializationUtils class method) deterministic (gigl.common.data.dataloaders.TFDatasetOptions attribute) DirectAU (class in gigl.src.common.models.layers.task) DIRECTORY (gigl.common.utils.local_fs.FileSystemEntity attribute) disk_size_gb (snapchat.research.gbml.gigl_resource_config_pb2.DataflowResourceConfig attribute) DISK_SIZE_GB_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.DataflowResourceConfig attribute) display_name (gigl.common.types.wrappers.kfp_api.KfpTaskDetails property) DistABLPLoader (class in gigl.distributed.distributed_neighborloader) DistLinkPredictionDataset (class in gigl.distributed) (class in gigl.distributed.dist_link_prediction_dataset) DistNeighborLoader (class in gigl.distributed) (class in gigl.distributed.distributed_neighborloader) DistPartitioner (class in gigl.distributed) (class in gigl.distributed.dist_partitioner) DistRangePartitioner (class in gigl.distributed) (class in gigl.distributed.dist_range_partitioner) DistributedContext (class in gigl.distributed) (class in gigl.distributed.dist_context) DistributedTrainerConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) DOCKER_LATEST_BASE_CPU_IMAGE_NAME_WITH_TAG (in module gigl.common.constants) DOCKER_LATEST_BASE_CUDA_IMAGE_NAME_WITH_TAG (in module gigl.common.constants) DOCKER_LATEST_BASE_DATAFLOW_IMAGE_NAME_WITH_TAG (in module gigl.common.constants) does_bq_table_exist() (gigl.src.common.utils.bq.BqUtils method) does_cluster_exist() (gigl.common.services.dataproc.DataprocService method) does_edge_table_have_dangling_edges() (gigl.analytics.graph_validation.bq_graph_validator.BQGraphValidator static method) (gigl.analytics.graph_validation.BQGraphValidator static method) does_gcs_file_exist() (gigl.common.utils.gcs.GcsUtils method) does_path_exist() (in module gigl.common.utils.local_fs) does_uri_exist() (gigl.src.common.utils.file_loader.FileLoader method) download() (gigl.src.mocking.lib.pyg_datasets_forks.CoraFromGCS method) (gigl.src.mocking.lib.pyg_datasets_forks.DBLPFromGCS method) download_file_from_gcs() (gigl.common.utils.gcs.GcsUtils method) download_file_from_gcs_to_temp_file() (gigl.common.utils.gcs.GcsUtils method) download_files_from_gcs_paths_to_local_dir() (gigl.common.utils.gcs.GcsUtils method) download_files_from_gcs_paths_to_local_paths() (gigl.common.utils.gcs.GcsUtils method) drop_feature() (in module gigl.src.common.models.pyg.graph.augmentations) dropout (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) dst_identifier (gigl.src.data_preprocessor.lib.ingest.reference.EdgeDataReference attribute) dst_node (gigl.src.common.types.graph_data.Edge property) (gigl.src.data_preprocessor.lib.types.EdgeOutputIdentifier attribute) dst_node_id (gigl.src.common.types.graph_data.Edge attribute) (gigl.src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper property) (snapchat.research.gbml.graph_schema_pb2.Edge attribute) DST_NODE_ID_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.Edge attribute) dst_node_id_key (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput attribute) DST_NODE_ID_KEY_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput attribute) dst_node_type (gigl.src.common.types.graph_data.EdgeType attribute) (snapchat.research.gbml.graph_schema_pb2.EdgeType attribute) DST_NODE_TYPE_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.EdgeType attribute) dtype (gigl.common.utils.compute.serialization.serialize_np.EncodedNdArray attribute) E early_stop_patience (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec attribute) EarlyStopper (class in gigl.src.common.modeling_task_specs.utils.early_stop) Edge (class in gigl.src.common.types.graph_data) (class in snapchat.research.gbml.graph_schema_pb2) EDGE (gigl.src.common.types.features.FeatureTypes attribute) edge_data (gigl.src.data_preprocessor.data_preprocessor.PreprocessedMetadataReferences attribute) edge_dir (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) edge_drop (gigl.src.common.models.layers.task.FeatureReconstruction attribute) edge_drop_1 (gigl.src.common.models.layers.task.BGRL attribute) (gigl.src.common.models.layers.task.GBT attribute) (gigl.src.common.models.layers.task.GRACE attribute) (gigl.src.common.models.layers.task.TBGRL attribute) (gigl.src.common.models.layers.task.WhiteningDecorrelation attribute) edge_drop_2 (gigl.src.common.models.layers.task.BGRL attribute) (gigl.src.common.models.layers.task.GBT attribute) (gigl.src.common.models.layers.task.GRACE attribute) (gigl.src.common.models.layers.task.TBGRL attribute) (gigl.src.common.models.layers.task.WhiteningDecorrelation attribute) edge_drop_neg (gigl.src.common.models.layers.task.TBGRL attribute) edge_dst_column_name (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) edge_entity_info (gigl.common.data.load_torch_tensors.SerializedGraphMetadata attribute) edge_feat_name (gigl.src.common.types.pb_wrappers.sampling_op.RandomWeightedPbWrapper property) (gigl.src.common.types.pb_wrappers.sampling_op.TopKPbWrapper property) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.RandomWeighted attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.TopK attribute) EDGE_FEAT_NAME_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.RandomWeighted attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.TopK attribute) edge_feats (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite.ToyGraphData attribute) (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) edge_features (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) (gigl.types.graph.LoadedGraphTensors attribute) EDGE_FEATURES_GRAPH_EDGELIST_ENUMERATION_QUERY (in module gigl.src.data_preprocessor.lib.enumerate.queries) edge_from_EdgePb() (gigl.src.common.translators.gbml_protos_translator.GbmlProtosTranslator static method) edge_ids (gigl.types.graph.GraphPartitionData attribute) edge_in_dim (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) edge_index (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) (gigl.types.graph.GraphPartitionData attribute) (gigl.types.graph.LoadedGraphTensors attribute) edge_indices (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite.ToyGraphData attribute) edge_partition_book (gigl.types.graph.PartitionOutput attribute) edge_pb (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) edge_preprocessor_config (snapchat.research.gbml.gigl_resource_config_pb2.DataPreprocessorConfig property) EDGE_PREPROCESSOR_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.DataPreprocessorConfig attribute) edge_src_column_name (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) edge_type (gigl.src.common.types.graph_data.Edge attribute) (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper property) (gigl.src.data_preprocessor.lib.ingest.reference.EdgeDataReference attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp property) edge_type_dim (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) edge_type_emb (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) EDGE_TYPE_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp attribute) edge_type_from_EdgeTypePb() (gigl.src.common.translators.gbml_protos_translator.GbmlProtosTranslator static method) edge_type_lin_dict (gigl.src.common.models.pyg.heterogeneous.SimpleHGN attribute) edge_type_to_condensed_edge_type_map (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) edge_types (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite.ToyGraphData attribute) (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo property) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata property) EDGE_TYPES_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.GraphMetadata attribute) edge_types_map (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) edge_types_to_be_registered (gigl.src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol property) (gigl.src.common.graph_builder.pyg_graph_data.PygGraphData property) edge_usage_type (gigl.src.data_preprocessor.lib.ingest.reference.EdgeDataReference attribute) EdgeAttrGAT (class in gigl.src.common.models.pyg.homogeneous) EdgeAttrGATConv (class in gigl.src.common.models.pyg.nn.conv.edge_attr_gat_conv) EdgeDataPreprocessingSpec (class in gigl.src.data_preprocessor.lib.types) EdgeDataReference (class in gigl.src.data_preprocessor.lib.ingest.reference) EdgeOutputIdentifier (class in gigl.src.data_preprocessor.lib.types) EdgePbWrapper (class in gigl.src.common.types.pb_wrappers.graph_data_types) EdgePbWrapperCoder (class in gigl.common.utils.compute.serialization.serialize_protos) edges (snapchat.research.gbml.graph_schema_pb2.Graph property) EDGES_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.Graph attribute) edges_pb (gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper property) edges_wrapper (gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper property) EdgeType (class in gigl.src.common.types.graph_data) (class in snapchat.research.gbml.graph_schema_pb2) EdgeTypePb_from_edge_type() (gigl.src.common.translators.gbml_protos_translator.GbmlProtosTranslator static method) EdgeUsageType (class in gigl.src.common.types.graph_data) embedding_bq_dataset_name (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) EMBEDDING_BQ_DATASET_NAME_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) EMBEDDING_TAGGED_OUTPUT_KEY (in module gigl.src.inference.v1.lib.base_inference_blueprint) EmbeddingExporter (class in gigl.common.data.export) embeddings (gigl.src.inference.v1.lib.base_inferencer.InferBatchResults attribute) embeddings_path (snapchat.research.gbml.inference_metadata_pb2.InferenceOutput attribute) EMBEDDINGS_PATH_FIELD_NUMBER (snapchat.research.gbml.inference_metadata_pb2.InferenceOutput attribute) enable_web_access (gigl.common.services.vertex_ai.VertexAiJobConfig attribute) encode() (gigl.common.beam.coders.PassthroughCoder method) (gigl.common.utils.compute.serialization.coder.CoderProtocol method) (gigl.common.utils.compute.serialization.serialize_np.NumpyCoder method) (gigl.common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (gigl.common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (gigl.common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) (gigl.common.utils.compute.serialization.serialize_string.StringCoder method) encode_value_as_feature() (gigl.src.data_preprocessor.lib.transform.tf_value_encoder.TFValueEncoder static method) EncodedNdArray (class in gigl.common.utils.compute.serialization.serialize_np) encoder (gigl.src.common.models.layers.task.BGRL attribute) (gigl.src.common.models.layers.task.FeatureReconstruction attribute) (gigl.src.common.models.layers.task.GBT attribute) (gigl.src.common.models.layers.task.GRACE attribute) (gigl.src.common.models.layers.task.TBGRL attribute) (gigl.src.common.models.layers.task.WhiteningDecorrelation attribute) encoding (gigl.common.utils.compute.serialization.serialize_string.StringCoder attribute) entity_key (gigl.common.data.dataloaders.SerializedTFRecordInfo attribute) entity_type (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) enumerated_edge_data_bq_table (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) ENUMERATED_EDGE_DATA_BQ_TABLE_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) enumerated_edge_data_reference (gigl.src.data_preprocessor.lib.enumerate.utils.EnumeratorEdgeTypeMetadata attribute) enumerated_node_data_bq_table (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) ENUMERATED_NODE_DATA_BQ_TABLE_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) enumerated_node_data_reference (gigl.src.data_preprocessor.lib.enumerate.utils.EnumeratorNodeTypeMetadata attribute) enumerated_node_ids_bq_table (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) ENUMERATED_NODE_IDS_BQ_TABLE_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) Enumerator (class in gigl.src.data_preprocessor.lib.enumerate.utils) EnumeratorEdgeTypeMetadata (class in gigl.src.data_preprocessor.lib.enumerate.utils) EnumeratorNodeTypeMetadata (class in gigl.src.data_preprocessor.lib.enumerate.utils) environment_variables (gigl.common.services.vertex_ai.VertexAiJobConfig attribute) eps (gigl.src.common.models.layers.loss.GBTLoss attribute) error_type (gigl.src.common.types.exception.SubgraphSamplingValidationError attribute) estimate() (gigl.src.common.models.layers.count_min_sketch.CountMinSketch method) estimate_torch_long_tensor() (gigl.src.common.models.layers.count_min_sketch.CountMinSketch method) eval() (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec method) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec method) (gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec.NodeClassificationModelingTaskSpec method) (gigl.src.training.v1.lib.base_trainer.BaseTrainer method) eval_metrics_uri (snapchat.research.gbml.trained_model_metadata_pb2.TrainedModelMetadata attribute) EVAL_METRICS_URI_FIELD_NUMBER (snapchat.research.gbml.trained_model_metadata_pb2.TrainedModelMetadata attribute) EvalMetric (class in gigl.src.common.types.model_eval_metrics) EvalMetricsCollection (class in gigl.src.common.types.model_eval_metrics) EvalMetricsCollectionTranslator (class in gigl.src.common.translators.model_eval_metrics_translator) EvalMetricType (class in gigl.src.common.types.model_eval_metrics) EXAMPLE_TASK_ASSETS_GCS_PATH (in module gigl.src.mocking.lib.constants) expand() (gigl.common.beam.better_tfrecordio.BetterWriteToTFRecord method) (gigl.src.data_preprocessor.lib.transform.utils.AnalyzeAndBuildTFTransformFn method) (gigl.src.data_preprocessor.lib.transform.utils.GenerateAndVisualizeStats method) (gigl.src.data_preprocessor.lib.transform.utils.IngestRawFeatures method) (gigl.src.data_preprocessor.lib.transform.utils.ReadExistingTFTransformFn method) (gigl.src.data_preprocessor.lib.transform.utils.WriteTFSchema method) (gigl.src.data_preprocessor.lib.types.InstanceDictPTransform method) (gigl.src.inference.v1.lib.utils.UnenumerateAssets method) experimental_flags (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig property) EXPERIMENTAL_FLAGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) export_to_gcs() (gigl.src.common.utils.bq.BqUtils method) F F (in module gigl.common.metrics.decorators) facets_report_uri (gigl.src.data_preprocessor.lib.transform.utils.GenerateAndVisualizeStats attribute) feat_drop_1 (gigl.src.common.models.layers.task.BGRL attribute) (gigl.src.common.models.layers.task.GBT attribute) (gigl.src.common.models.layers.task.GRACE attribute) (gigl.src.common.models.layers.task.TBGRL attribute) (gigl.src.common.models.layers.task.WhiteningDecorrelation attribute) feat_drop_2 (gigl.src.common.models.layers.task.BGRL attribute) (gigl.src.common.models.layers.task.GBT attribute) (gigl.src.common.models.layers.task.GRACE attribute) (gigl.src.common.models.layers.task.TBGRL attribute) (gigl.src.common.models.layers.task.WhiteningDecorrelation attribute) feat_drop_neg (gigl.src.common.models.layers.task.TBGRL attribute) feats (gigl.types.graph.FeaturePartitionData attribute) feats_interaction (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) feature_dim (gigl.common.data.dataloaders.SerializedTFRecordInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) FEATURE_DIM_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) feature_dim_output (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) feature_embedding_layer (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) feature_embedding_layers (gigl.src.common.models.pyg.heterogeneous.HGT attribute) feature_flags (snapchat.research.gbml.gbml_config_pb2.GbmlConfig property) FEATURE_FLAGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig attribute) feature_index (gigl.src.data_preprocessor.lib.types.FeatureSchema attribute) feature_keys (gigl.common.data.dataloaders.SerializedTFRecordInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo property) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput property) FEATURE_KEYS_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) feature_spec (gigl.common.data.dataloaders.SerializedTFRecordInfo attribute) (gigl.src.data_preprocessor.lib.transform.utils.IngestRawFeatures attribute) (gigl.src.data_preprocessor.lib.transform.utils.InstanceDictToTFExample attribute) (gigl.src.data_preprocessor.lib.types.FeatureSchema attribute) feature_spec_fn (gigl.src.data_preprocessor.lib.types.EdgeDataPreprocessingSpec attribute) (gigl.src.data_preprocessor.lib.types.NodeDataPreprocessingSpec attribute) feature_spec_to_feature_index_map() (in module gigl.common.utils.tensorflow_schema) feature_type (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) feature_values (gigl.src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper property) (gigl.src.common.types.pb_wrappers.graph_data_types.NodePbWrapper property) (snapchat.research.gbml.graph_schema_pb2.Edge property) (snapchat.research.gbml.graph_schema_pb2.Node property) FEATURE_VALUES_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.Edge attribute) (snapchat.research.gbml.graph_schema_pb2.Node attribute) feature_vocab (gigl.src.data_preprocessor.lib.types.FeatureSchema attribute) FeatureEmbeddingLayer (class in gigl.src.common.models.pyg.nn.models.feature_embedding) FeatureIndexDict (in module gigl.src.data_preprocessor.lib.types) FeatureInteraction (class in gigl.src.common.models.pyg.nn.models.feature_interaction) FeaturePartitionData (class in gigl.types.graph) FeatureReconstruction (class in gigl.src.common.models.layers.task) FeatureReconstructionLoss (class in gigl.src.common.models.layers.loss) features_outputs (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) (gigl.src.data_preprocessor.lib.types.EdgeDataPreprocessingSpec attribute) (gigl.src.data_preprocessor.lib.types.NodeDataPreprocessingSpec attribute) FeatureSchema (class in gigl.src.data_preprocessor.lib.types) FeatureSchemaDict (in module gigl.src.data_preprocessor.lib.types) FeatureSerializationUtils (class in gigl.src.common.utils.data.feature_serialization) FeatureSpecDict (in module gigl.src.data_preprocessor.lib.types) FeatureTypes (class in gigl.src.common.types.features) FeatureVocabDict (in module gigl.src.data_preprocessor.lib.types) fetch_bq_table_schema() (gigl.src.common.utils.bq.BqUtils method) FILE (gigl.common.utils.local_fs.FileSystemEntity attribute) file_buffer_size (gigl.common.data.dataloaders.TFDatasetOptions attribute) file_loader (in module gigl.src.validation_check.libs.frozen_config_path_checks) FileLoader (class in gigl.src.common.utils.file_loader) FileSystemEntity (class in gigl.common.utils.local_fs) filter_dict() (in module gigl.src.common.models.pyg.utils) filter_features() (in module gigl.src.common.utils.data.training) finished_at (gigl.common.types.wrappers.kfp_api.ApiRunDetailWrapper property) (gigl.common.types.wrappers.kfp_api.KfpTaskDetails property) flattened_graph_metadata (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig property) FLATTENED_GRAPH_METADATA_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) flattened_graph_metadata_pb (gigl.src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper attribute) flattened_graph_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) FLATTENED_GRAPH_TO_DATASET_TYPE (in module gigl.src.common.types.pb_wrappers.flattened_graph_metadata) FLATTENED_GRAPH_TO_TRAINING_SAMPLE_TYPE (in module gigl.src.common.types.pb_wrappers.flattened_graph_metadata) FlattenedGraphMetadata (class in snapchat.research.gbml.flattened_graph_metadata_pb2) FlattenedGraphMetadataOutputPb (in module gigl.src.common.types.pb_wrappers.types) FlattenedGraphMetadataPbWrapper (class in gigl.src.common.types.pb_wrappers.flattened_graph_metadata) flip_edge() (gigl.src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper method) flush_embeddings() (gigl.common.data.export.EmbeddingExporter method) flush_metrics() (gigl.common.metrics.base_metrics.NopMetricsPublisher method) (gigl.common.metrics.metrics_interface.OpsMetricPublisher method) flushes_metrics() (in module gigl.common.metrics.decorators) format_bq_path() (gigl.src.common.utils.bq.BqUtils static method) format_datetime() (in module gigl.src.common.utils.time) forward() (gigl.src.common.models.layers.decoder.LinkPredictionDecoder method) (gigl.src.common.models.layers.feature_interaction.DCNCross method) (gigl.src.common.models.layers.feature_interaction.DCNv2 method) (gigl.src.common.models.layers.loss.AligmentLoss method) (gigl.src.common.models.layers.loss.BGRLLoss method) (gigl.src.common.models.layers.loss.FeatureReconstructionLoss method) (gigl.src.common.models.layers.loss.GBTLoss method) (gigl.src.common.models.layers.loss.GRACELoss method) (gigl.src.common.models.layers.loss.KLLoss method) (gigl.src.common.models.layers.loss.LLPRankingLoss method) (gigl.src.common.models.layers.loss.MarginLoss method) (gigl.src.common.models.layers.loss.RetrievalLoss method) (gigl.src.common.models.layers.loss.SoftmaxLoss method) (gigl.src.common.models.layers.loss.TBGRLLoss method) (gigl.src.common.models.layers.loss.UniformityLoss method) (gigl.src.common.models.layers.loss.WhiteningDecorrelationLoss method) (gigl.src.common.models.layers.task.BGRL method) (gigl.src.common.models.layers.task.DirectAU method) (gigl.src.common.models.layers.task.FeatureReconstruction method) (gigl.src.common.models.layers.task.GBT method) (gigl.src.common.models.layers.task.GRACE method) (gigl.src.common.models.layers.task.Margin method) (gigl.src.common.models.layers.task.NodeAnchorBasedLinkPredictionBaseTask method) (gigl.src.common.models.layers.task.Retrieval method) (gigl.src.common.models.layers.task.Softmax method) (gigl.src.common.models.layers.task.TBGRL method) (gigl.src.common.models.layers.task.WhiteningDecorrelation method) (gigl.src.common.models.pyg.heterogeneous.HGT method) (gigl.src.common.models.pyg.heterogeneous.SimpleHGN method) (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN method) (gigl.src.common.models.pyg.homogeneous.TwoLayerGCN method) (gigl.src.common.models.pyg.link_prediction.LinkPredictionGNN method) (gigl.src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (gigl.src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (gigl.src.common.models.pyg.nn.models.feature_embedding.FeatureEmbeddingLayer method) (gigl.src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (gigl.src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) from_dict() (gigl.src.mocking.lib.versioning.MockedDatasetArtifactMetadata static method) from_eval_metric_type() (gigl.src.common.types.model_eval_metrics.EvalMetric class method) from_hetero_data() (gigl.src.common.graph_builder.pyg_graph_data.PygGraphData class method) from_nodes() (gigl.src.common.types.graph_data.Edge class method) from_serialized() (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper class method) from_string() (gigl.orchestration.kubeflow.runner.Action static method) frozen_gbml_config_uri (gigl.src.mocking.lib.versioning.MockedDatasetArtifactMetadata attribute) FrozenDict (class in gigl.common.collections.frozen_dict) G gamma (gigl.src.common.models.layers.task.DirectAU attribute) GAT (class in gigl.src.common.models.pyg.homogeneous) GATv2 (class in gigl.src.common.models.pyg.homogeneous) GAUGE_MODEL_METRICS_LOSS (in module gigl.src.training.v1.lib.eval_metrics) GAUGE_MODEL_METRICS_MRR (in module gigl.src.training.v1.lib.eval_metrics) gauges (gigl.common.metrics.base_metrics.NopMetricsPublisher attribute) gbml_config_pb (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper attribute) gbml_config_pb_wrapper (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec property) (gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec.NodeClassificationModelingTaskSpec property) (gigl.src.data_preprocessor.data_preprocessor.DataPreprocessor property) (gigl.src.inference.v1.gnn_inferencer.InferencerV1 property) GbmlConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.DatasetConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.DatasetConfig.DataPreprocessorConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.DatasetConfig.SplitGeneratorConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.DatasetConfig.SubgraphSamplerConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.FeatureFlagsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.GraphDBConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.GraphDBConfig.GraphDbArgsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.GraphDBConfig.GraphDBServiceConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.InferencerConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.InferencerConfig.InferencerArgsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.MetricsConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.MetricsConfig.MetricsArgsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.PostProcessorConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.ProfilerConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.ProfilerConfig.ProfilerArgsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.SharedConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.SharedConfig.SharedArgsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.TaskMetadata (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.TaskMetadata.LinkBasedTaskMetadata (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.TaskMetadata.NodeBasedTaskMetadata (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.TrainerConfig (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfig.TrainerConfig.TrainerArgsEntry (class in snapchat.research.gbml.gbml_config_pb2) GbmlConfigPbWrapper (class in gigl.src.common.types.pb_wrappers.gbml_config) GbmlGraphDataProtocol (class in gigl.src.common.graph_builder.gbml_graph_protocol) GbmlProtosTranslator (class in gigl.src.common.translators.gbml_protos_translator) GBT (class in gigl.src.common.models.layers.task) GBTLoss (class in gigl.src.common.models.layers.loss) gcp_service_account_email (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) GCP_SERVICE_ACCOUNT_EMAIL_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) GCPCloudLoggingService (class in gigl.common.services.gcp_cloud_logging) GcsUri (class in gigl.common) (class in gigl.common.types.uri.gcs_uri) GcsUtils (class in gigl.common.utils.gcs) generate_bigquery_assets() (in module gigl.src.mocking.lib.mock_input_for_data_preprocessor) generate_inferencer_instance() (gigl.src.inference.v1.gnn_inferencer.InferencerV1 method) generate_pipeline() (in module gigl.orchestration.kubeflow.kfp_pipeline) generate_preprocessed_metadata_pb() (gigl.src.data_preprocessor.data_preprocessor.DataPreprocessor method) generate_preprocessed_tfrecord_data() (in module gigl.src.mocking.lib.mock_input_for_subgraph_sampler) generate_trainer_instance() (in module gigl.src.training.v1.lib.training_process) GenerateAndVisualizeStats (class in gigl.src.data_preprocessor.lib.transform.utils) get() (gigl.src.common.utils.scheduler.CosineDecayScheduler method) get_all_criteria() (gigl.src.common.models.layers.decoder.DecoderType class method) (gigl.src.common.types.model_eval_metrics.EvalMetricType class method) get_all_directions() (gigl.src.common.types.pb_wrappers.sampling_op.SamplingDirectionEnumType class method) get_applied_task_perm_gcs_path() (in module gigl.src.common.constants.gcs) get_applied_task_temp_gcs_path() (in module gigl.src.common.constants.gcs) get_applied_task_temp_regional_gcs_path() (in module gigl.src.common.constants.gcs) get_augmented_graph() (in module gigl.src.common.models.pyg.graph.augmentations) get_available_cpus() (in module gigl.common.env_config) get_available_device() (in module gigl.distributed.utils) (in module gigl.distributed.utils.device) get_basename() (gigl.common.types.uri.uri.Uri method) (gigl.common.Uri method) get_batch_generator_fn() (gigl.src.inference.v1.lib.base_inference_blueprint.BaseInferenceBlueprint method) (gigl.src.inference.v1.lib.node_anchor_based_link_prediction_inferencer.NodeAnchorBasedLinkPredictionInferenceBlueprint method) (gigl.src.inference.v1.lib.node_classification_inferencer.NodeClassificationInferenceBlueprint method) get_bucket_and_blob_path_from_gcs_path() (gigl.common.utils.gcs.GcsUtils static method) get_compiled_jar_path() (in module gigl.env.dep_constants) get_config_populator_assets_perm_gcs_path() (in module gigl.src.common.constants.gcs) get_console_uri_from_pipeline_result() (in module gigl.common.utils.dataflow) get_current_jar_file() (in module gigl.env.dep_constants) get_data_preprocessor_assets_perm_gcs_path() (in module gigl.src.common.constants.gcs) get_data_preprocessor_assets_temp_gcs_path() (in module gigl.src.common.constants.gcs) get_data_preprocessor_staging_gcs_path() (in module gigl.src.common.constants.gcs) get_data_split_for_current_worker() (in module gigl.src.training.v1.lib.data_loaders.utils) get_dataflow_staging_gcs_path() (in module gigl.src.common.constants.gcs) get_dataflow_temp_gcs_path() (in module gigl.src.common.constants.gcs) get_dataset_name_from_table() (gigl.src.common.utils.bq.BqUtils method) get_default_data_loader() (gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader.NodeAnchorBasedLinkPredictionBatch static method) (gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader.RootedNodeNeighborhoodBatch static method) (gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader.SupervisedNodeClassificationBatch static method) get_dehydrated_node_pb_wrappers_from_edge_wrapper() (in module gigl.src.common.types.pb_wrappers.graph_data_types_utils) get_distributed_backend() (in module gigl.common.utils.torch_training) get_edges_preprocessing_spec() (gigl.src.data_preprocessor.lib.data_preprocessor_config.DataPreprocessorConfig method) (gigl.src.mocking.mocking_assets.passthrough_preprocessor_config_for_mocked_assets.PassthroughPreprocessorConfigForMockedAssets method) get_emb_table_schema() (gigl.src.inference.v1.lib.base_inference_blueprint.BaseInferenceBlueprint static method) get_embeddings_dataset_bq_path() (in module gigl.src.common.constants.bq) get_embeddings_table() (in module gigl.src.common.constants.bq) get_enumerated_edge_features_bq_table_name() (in module gigl.src.data_preprocessor.lib.enumerate.utils) get_enumerated_embedding_table_path() (gigl.src.inference.lib.assets.InferenceAssets static method) get_enumerated_node_features_bq_table_name() (in module gigl.src.data_preprocessor.lib.enumerate.utils) get_enumerated_node_id_map_bq_table_name() (in module gigl.src.data_preprocessor.lib.enumerate.utils) get_enumerated_predictions_table_path() (gigl.src.inference.lib.assets.InferenceAssets static method) get_example_task_edge_features_gcs_dir() (in module gigl.src.mocking.lib.constants) get_example_task_edge_features_schema_gcs_path() (in module gigl.src.mocking.lib.constants) get_example_task_edges_bq_table_path() (in module gigl.src.mocking.lib.constants) get_example_task_frozen_gbml_config_gcs_path() (in module gigl.src.mocking.lib.constants) get_example_task_node_features_gcs_dir() (in module gigl.src.mocking.lib.constants) get_example_task_node_features_schema_gcs_path() (in module gigl.src.mocking.lib.constants) get_example_task_nodes_bq_table_path() (in module gigl.src.mocking.lib.constants) get_example_task_preprocess_gcs_prefix() (in module gigl.src.mocking.lib.constants) get_example_task_static_assets_gcs_dir() (in module gigl.src.mocking.lib.constants) get_feature_field_name() (in module gigl.src.mocking.lib.feature_handling) get_feature_len_from_fixed_len_feature() (in module gigl.common.utils.tensorflow_schema) get_frozen_gbml_config_proto_gcs_path() (in module gigl.src.common.constants.gcs) get_gbml_assets_tmp_path() (in module gigl.src.common.constants.local_fs) get_gbml_config_pb_wrapper_from_uri() (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper class method) get_gbml_log_file_prefix_tmp_path() (in module gigl.src.common.constants.local_fs) get_gbml_logs_folder_local_tmp_path() (in module gigl.src.common.constants.local_fs) get_gbml_task_local_tmp_path() (in module gigl.src.common.constants.local_fs) get_gcs_asset_write_path_prefix() (gigl.src.inference.lib.assets.InferenceAssets static method) get_gigl_root_directory() (in module gigl.src.common.constants.local_fs) get_global_edge_features_dict() (gigl.src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol method) (gigl.src.common.graph_builder.pyg_graph_data.PygGraphData method) get_global_node_features_dict() (gigl.src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol method) (gigl.src.common.graph_builder.pyg_graph_data.PygGraphData method) get_graph_builder() (gigl.src.common.graph_builder.graph_builder_factory.GraphBuilderFactory class method) get_host_name() (in module gigl.common.utils.vertex_ai_context) get_ids_on_rank() (in module gigl.distributed.utils.partition_book) get_inference_blueprint() (gigl.src.inference.v1.lib.inference_blueprint_factory.InferenceBlueprintFactory class method) get_inference_data_tf_record_uri_prefixes() (gigl.src.inference.v1.lib.base_inference_blueprint.BaseInferenceBlueprint method) (gigl.src.inference.v1.lib.node_anchor_based_link_prediction_inferencer.NodeAnchorBasedLinkPredictionInferenceBlueprint method) (gigl.src.inference.v1.lib.node_classification_inferencer.NodeClassificationInferenceBlueprint method) get_inference_embeddings_local_tmp_dir_path() (in module gigl.src.common.constants.local_fs) get_inference_embeddings_local_tmp_file_prefix() (in module gigl.src.common.constants.local_fs) get_inference_predictions_local_tmp_file_prefix() (in module gigl.src.common.constants.local_fs) get_inferencer_asset_dir_gcs_path() (in module gigl.src.common.constants.gcs) get_inferencer_embeddings_gcs_prefix() (in module gigl.src.common.constants.gcs) get_inferencer_pipeline_component_for_single_node_type() (in module gigl.src.inference.v1.lib.utils) get_inferencer_predictions_gcs_prefix() (in module gigl.src.common.constants.gcs) get_inferer() (gigl.src.inference.v1.lib.base_inference_blueprint.BaseInferenceBlueprint method) get_jar_file_uri() (in module gigl.env.dep_constants) get_labels_for_anchor_nodes() (in module gigl.utils.data_splitters) get_leader_hostname() (in module gigl.common.utils.vertex_ai_context) get_leader_port() (in module gigl.common.utils.vertex_ai_context) get_load_data_and_transform_pipeline_component() (in module gigl.src.data_preprocessor.lib.transform.utils) get_local_jar_directory() (in module gigl.env.dep_constants) get_logs_iterator_from_k8_container() (gigl.common.services.gcp_cloud_logging.GCPCloudLoggingService method) get_metric_name_for_gauge_model_metrics_hit_rate_at() (in module gigl.src.training.v1.lib.eval_metrics) get_metrics_service_instance() (in module gigl.src.common.utils.metrics_service_provider) get_mocked_dataset_artifact_metadata() (in module gigl.src.mocking.lib.versioning) get_nodes_preprocessing_spec() (gigl.src.data_preprocessor.lib.data_preprocessor_config.DataPreprocessorConfig method) (gigl.src.mocking.mocking_assets.passthrough_preprocessor_config_for_mocked_assets.PassthroughPreprocessorConfigForMockedAssets method) get_np_iterator_from_tfrecords() (in module gigl.src.training.v1.lib.data_loaders.tf_records_iterable_dataset) get_num_edges() (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo method) get_output_paths() (gigl.src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper method) (gigl.src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper method) (gigl.src.common.types.pb_wrappers.trained_model_metadata.TrainedModelMetadataPbWrapper method) get_path_to_manifest_file() (in module gigl.src.common.constants.local_fs) get_pipeline_job_from_job_name() (gigl.common.services.vertex_ai.VertexAIService method) get_pipeline_run_url() (gigl.common.services.vertex_ai.VertexAIService static method) get_post_processor_asset_dir_gcs_path() (in module gigl.src.common.constants.gcs) get_post_processor_metrics_gcs_path() (in module gigl.src.common.constants.gcs) get_pred_table_schema() (gigl.src.inference.v1.lib.base_inference_blueprint.BaseInferenceBlueprint static method) get_predictions_table() (in module gigl.src.common.constants.bq) get_preprocess_local_staging_dir_path() (in module gigl.src.common.constants.local_fs) get_preprocessed_metadata_proto_gcs_path() (in module gigl.src.common.constants.gcs) get_process_group_name() (in module gigl.distributed.utils) (in module gigl.distributed.utils.init_neighbor_loader_worker) get_project_root_directory() (in module gigl.src.common.constants.local_fs) get_pyg_cora_dataset() (in module gigl.src.applied_tasks.test_tasks.academic) get_python_project_root_path() (in module gigl.src.common.constants.local_fs) get_rank() (in module gigl.common.utils.torch_training) (in module gigl.common.utils.vertex_ai_context) get_resource_config() (in module gigl.env.pipelines_config) get_resource_config_uri (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) get_resource_labels() (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper method) (in module gigl.src.data_preprocessor.lib.enumerate.utils) get_resource_labels_formatted_for_dataflow() (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper method) get_running_job_ids_on_cluster() (gigl.common.services.dataproc.DataprocService method) get_sanitized_dataflow_job_name() (in module gigl.src.common.utils.dataflow) get_sanitized_dataproc_cluster_name() (gigl.src.common.utils.spark_job_manager.SparkJobManager static method) get_split_dataset_main_samples_gcs_file_prefix() (in module gigl.src.common.constants.gcs) get_split_dataset_output_gcs_file_prefix() (in module gigl.src.common.constants.gcs) get_split_dataset_random_negatives_gcs_file_prefix() (in module gigl.src.common.constants.gcs) get_split_generator_assets_temp_gcs_path() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_flattened_graph_metadata_output_gcs_path() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_node_anchor_based_link_prediction_random_negatives_samples_prefix() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_node_anchor_based_link_prediction_samples_prefix() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_node_anchor_based_link_prediction_task_dir() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_node_neighborhood_samples_dir() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_node_neighborhood_samples_path_prefix() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_root_dir() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_supervised_link_based_task_dir() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_supervised_link_based_task_labeled_samples_prefix() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_supervised_link_based_task_unlabeled_samples_prefix() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_supervised_node_classification_labeled_samples_prefix() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_supervised_node_classification_task_dir() (in module gigl.src.common.constants.gcs) get_subgraph_sampler_supervised_node_classification_unlabeled_samples_prefix() (in module gigl.src.common.constants.gcs) get_submitted_job_ids() (gigl.common.services.dataproc.DataprocService method) get_supervision_edge_node_types() (gigl.src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper method) get_supervision_edge_types() (gigl.src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper method) get_supervision_node_types() (gigl.src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper method) get_table() (gigl.src.common.models.layers.count_min_sketch.CountMinSketch method) get_table_names_within_date_range() (gigl.src.common.utils.bq.BqUtils method) get_table_paths_via_timedelta() (in module gigl.src.post_process.utils.cosine_similarity) get_task_root_node_types() (gigl.src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper method) get_tensorboard_logs_gcs_path() (in module gigl.src.common.constants.gcs) get_test_dataloaders() (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.NodeAnchorBasedLinkPredictionDatasetDataloaders method) (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.SupervisedNodeClassificationDatasetDataloaders method) get_tf_record_coder() (gigl.src.inference.v1.lib.base_inference_blueprint.BaseInferenceBlueprint method) (gigl.src.inference.v1.lib.node_anchor_based_link_prediction_inferencer.NodeAnchorBasedLinkPredictionInferenceBlueprint method) (gigl.src.inference.v1.lib.node_classification_inferencer.NodeClassificationInferenceBlueprint method) get_tf_transform_directory_path() (in module gigl.src.common.constants.gcs) get_tf_transform_raw_data_schema_file_path() (in module gigl.src.common.constants.gcs) get_tf_transform_stats_directory_path() (in module gigl.src.common.constants.gcs) get_tf_transform_stats_file_path() (in module gigl.src.common.constants.gcs) get_tf_transform_temp_directory_path() (in module gigl.src.common.constants.gcs) get_tf_transform_visualized_facets_file_path() (in module gigl.src.common.constants.gcs) get_tf_transformed_features_schema_path() (in module gigl.src.common.constants.gcs) get_tf_transformed_features_transform_fn_assets_directory_path() (in module gigl.src.common.constants.gcs) get_torch_profiler_instance() (in module gigl.src.training.v1.lib.training_process) get_train_val_info_local_tmp_file_path() (in module gigl.src.common.constants.local_fs) get_trained_model_eval_metrics_gcs_path() (in module gigl.src.common.constants.gcs) get_trained_model_gcs_path() (in module gigl.src.common.constants.gcs) get_trained_model_metadata_proto_gcs_path() (in module gigl.src.common.constants.gcs) get_trained_models_dir_gcs_path() (in module gigl.src.common.constants.gcs) get_trained_scripted_model_gcs_path() (in module gigl.src.common.constants.gcs) get_trainer_asset_dir_gcs_path() (in module gigl.src.common.constants.gcs) get_training_dataloaders() (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.NodeAnchorBasedLinkPredictionDatasetDataloaders method) (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.SupervisedNodeClassificationDatasetDataloaders method) get_transformed_features_directory_path() (in module gigl.src.common.constants.gcs) get_transformed_features_file_prefix() (in module gigl.src.common.constants.gcs) get_unenumerated_embedding_table_path() (gigl.src.inference.lib.assets.InferenceAssets static method) get_unenumerated_prediction_table_path() (gigl.src.inference.lib.assets.InferenceAssets static method) get_value_to_impute() (gigl.src.data_preprocessor.lib.transform.tf_value_encoder.TFValueEncoder static method) get_vertex_ai_job_id() (in module gigl.common.utils.vertex_ai_context) get_world_size() (in module gigl.common.utils.torch_training) (in module gigl.common.utils.vertex_ai_context) gigl module gigl.analytics module gigl.analytics.graph_validation module gigl.analytics.graph_validation.bq_graph_validator module gigl.common module gigl.common.beam module gigl.common.beam.better_tfrecordio module gigl.common.beam.coders module gigl.common.collections module gigl.common.collections.frozen_dict module gigl.common.collections.itertools module gigl.common.collections.sorted_dict module gigl.common.constants module gigl.common.data module gigl.common.data.dataloaders module gigl.common.data.export module gigl.common.data.load_torch_tensors module gigl.common.env_config module gigl.common.logger module gigl.common.metrics module gigl.common.metrics.base_metrics module gigl.common.metrics.decorators module gigl.common.metrics.metrics_interface module gigl.common.services module gigl.common.services.dataproc module gigl.common.services.gcp_cloud_logging module gigl.common.services.vertex_ai module gigl.common.types module gigl.common.types.resource_config module gigl.common.types.uri module gigl.common.types.uri.gcs_uri module gigl.common.types.uri.http_uri module gigl.common.types.uri.local_uri module gigl.common.types.uri.uri module gigl.common.types.uri.uri_factory module gigl.common.types.wrappers module gigl.common.types.wrappers.kfp_api module gigl.common.utils module gigl.common.utils.compute module gigl.common.utils.compute.random module gigl.common.utils.compute.serialization module gigl.common.utils.compute.serialization.coder module gigl.common.utils.compute.serialization.serialize_np module gigl.common.utils.compute.serialization.serialize_protos module gigl.common.utils.compute.serialization.serialize_string module gigl.common.utils.dataflow module gigl.common.utils.decorator module gigl.common.utils.func_tools module gigl.common.utils.gcs module gigl.common.utils.jupyter_magics module gigl.common.utils.kfp module gigl.common.utils.local_fs module gigl.common.utils.os_utils module gigl.common.utils.proto_utils module gigl.common.utils.retry module gigl.common.utils.tensorflow_schema module gigl.common.utils.test_utils module gigl.common.utils.torch_training module gigl.common.utils.vertex_ai_context module gigl.distributed module gigl.distributed.constants module gigl.distributed.dataset_factory module gigl.distributed.dist_context module gigl.distributed.dist_link_prediction_dataset module gigl.distributed.dist_partitioner module gigl.distributed.dist_range_partitioner module gigl.distributed.distributed_neighborloader module gigl.distributed.utils module gigl.distributed.utils.device module gigl.distributed.utils.init_neighbor_loader_worker module gigl.distributed.utils.partition_book module gigl.distributed.utils.serialized_graph_metadata_translator module gigl.env module gigl.env.dep_constants module gigl.env.pipelines_config module gigl.orchestration module gigl.orchestration.kubeflow module gigl.orchestration.kubeflow.kfp_orchestrator module gigl.orchestration.kubeflow.kfp_pipeline module gigl.orchestration.kubeflow.runner module gigl.orchestration.kubeflow.utils module gigl.orchestration.kubeflow.utils.glt_backend module gigl.orchestration.kubeflow.utils.log_metrics module gigl.orchestration.kubeflow.utils.resource module gigl.orchestration.local module gigl.orchestration.local.runner module gigl.src module gigl.src.applied_tasks module gigl.src.applied_tasks.test_tasks module gigl.src.applied_tasks.test_tasks.academic module gigl.src.common module gigl.src.common.constants module gigl.src.common.constants.bq module gigl.src.common.constants.components module gigl.src.common.constants.gcs module gigl.src.common.constants.graph_metadata module gigl.src.common.constants.local_fs module gigl.src.common.constants.metrics module gigl.src.common.constants.resource_config module gigl.src.common.constants.time module gigl.src.common.constants.training module gigl.src.common.graph_builder module gigl.src.common.graph_builder.abstract_graph_builder module gigl.src.common.graph_builder.gbml_graph_protocol module gigl.src.common.graph_builder.graph_builder_factory module gigl.src.common.graph_builder.pyg_graph_builder module gigl.src.common.graph_builder.pyg_graph_data module gigl.src.common.modeling_task_specs module gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec module gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec module gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec module gigl.src.common.modeling_task_specs.utils module gigl.src.common.modeling_task_specs.utils.early_stop module gigl.src.common.modeling_task_specs.utils.infer module gigl.src.common.modeling_task_specs.utils.parameter_loading module gigl.src.common.modeling_task_specs.utils.profiler_wrapper module gigl.src.common.models module gigl.src.common.models.dgl module gigl.src.common.models.dgl.homogenous module gigl.src.common.models.layers module gigl.src.common.models.layers.count_min_sketch module gigl.src.common.models.layers.decoder module gigl.src.common.models.layers.feature_interaction module gigl.src.common.models.layers.loss module gigl.src.common.models.layers.normalization module gigl.src.common.models.layers.task module gigl.src.common.models.pyg module gigl.src.common.models.pyg.graph module gigl.src.common.models.pyg.graph.augmentations module gigl.src.common.models.pyg.heterogeneous module gigl.src.common.models.pyg.homogeneous module gigl.src.common.models.pyg.link_prediction module gigl.src.common.models.pyg.nn module gigl.src.common.models.pyg.nn.conv module gigl.src.common.models.pyg.nn.conv.edge_attr_gat_conv module gigl.src.common.models.pyg.nn.conv.gin_conv module gigl.src.common.models.pyg.nn.conv.hgt_conv module gigl.src.common.models.pyg.nn.conv.simplehgn_conv module gigl.src.common.models.pyg.nn.models module gigl.src.common.models.pyg.nn.models.feature_embedding module gigl.src.common.models.pyg.nn.models.feature_interaction module gigl.src.common.models.pyg.nn.models.jumping_knowledge module gigl.src.common.models.pyg.utils module gigl.src.common.models.utils module gigl.src.common.models.utils.torch module gigl.src.common.translators module gigl.src.common.translators.gbml_protos_translator module gigl.src.common.translators.model_eval_metrics_translator module gigl.src.common.translators.training_samples_protos_translator module gigl.src.common.types module gigl.src.common.types.dataflow_job_options module gigl.src.common.types.dataset_split module gigl.src.common.types.exception module gigl.src.common.types.features module gigl.src.common.types.graph_data module gigl.src.common.types.model module gigl.src.common.types.model_eval_metrics module gigl.src.common.types.pb_wrappers module gigl.src.common.types.pb_wrappers.dataset_metadata module gigl.src.common.types.pb_wrappers.dataset_metadata_utils module gigl.src.common.types.pb_wrappers.flattened_graph_metadata module gigl.src.common.types.pb_wrappers.gbml_config module gigl.src.common.types.pb_wrappers.gigl_resource_config module gigl.src.common.types.pb_wrappers.graph_data_types module gigl.src.common.types.pb_wrappers.graph_data_types_utils module gigl.src.common.types.pb_wrappers.graph_metadata module gigl.src.common.types.pb_wrappers.preprocessed_metadata module gigl.src.common.types.pb_wrappers.sampling_op module gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy module gigl.src.common.types.pb_wrappers.task_metadata module gigl.src.common.types.pb_wrappers.trained_model_metadata module gigl.src.common.types.pb_wrappers.types module gigl.src.common.types.task_inputs module gigl.src.common.types.task_metadata module gigl.src.common.utils module gigl.src.common.utils.bq module gigl.src.common.utils.data module gigl.src.common.utils.data.feature_serialization module gigl.src.common.utils.data.training module gigl.src.common.utils.dataflow module gigl.src.common.utils.eval_metrics module gigl.src.common.utils.file_loader module gigl.src.common.utils.gbml_protos module gigl.src.common.utils.metrics_service_provider module gigl.src.common.utils.model module gigl.src.common.utils.scheduler module gigl.src.common.utils.spark_job_manager module gigl.src.common.utils.time module gigl.src.common.utils.timeout module gigl.src.config_populator module gigl.src.config_populator.config_populator module gigl.src.data_preprocessor module gigl.src.data_preprocessor.data_preprocessor module gigl.src.data_preprocessor.lib module gigl.src.data_preprocessor.lib.data_preprocessor_config module gigl.src.data_preprocessor.lib.enumerate module gigl.src.data_preprocessor.lib.enumerate.queries module gigl.src.data_preprocessor.lib.enumerate.utils module gigl.src.data_preprocessor.lib.ingest module gigl.src.data_preprocessor.lib.ingest.bigquery module gigl.src.data_preprocessor.lib.ingest.reference module gigl.src.data_preprocessor.lib.transform module gigl.src.data_preprocessor.lib.transform.tf_value_encoder module gigl.src.data_preprocessor.lib.transform.transformed_features_info module gigl.src.data_preprocessor.lib.transform.utils module gigl.src.data_preprocessor.lib.types module gigl.src.inference module gigl.src.inference.inferencer module gigl.src.inference.lib module gigl.src.inference.lib.assets module gigl.src.inference.v1 module gigl.src.inference.v1.gnn_inferencer module gigl.src.inference.v1.lib module gigl.src.inference.v1.lib.base_inference_blueprint module gigl.src.inference.v1.lib.base_inferencer module gigl.src.inference.v1.lib.inference_blueprint_factory module gigl.src.inference.v1.lib.inference_output_schema module gigl.src.inference.v1.lib.node_anchor_based_link_prediction_inferencer module gigl.src.inference.v1.lib.node_classification_inferencer module gigl.src.inference.v1.lib.queries module gigl.src.inference.v1.lib.transforms module gigl.src.inference.v1.lib.transforms.batch_generator module gigl.src.inference.v1.lib.transforms.utils module gigl.src.inference.v1.lib.utils module gigl.src.inference.v2 module gigl.src.inference.v2.glt_inferencer module gigl.src.mocking module gigl.src.mocking.dataset_asset_mocker module gigl.src.mocking.dataset_asset_mocking_suite module gigl.src.mocking.lib module gigl.src.mocking.lib.constants module gigl.src.mocking.lib.feature_handling module gigl.src.mocking.lib.gbml_config module gigl.src.mocking.lib.mock_input_for_data_preprocessor module gigl.src.mocking.lib.mock_input_for_inference module gigl.src.mocking.lib.mock_input_for_split_generator module gigl.src.mocking.lib.mock_input_for_subgraph_sampler module gigl.src.mocking.lib.mock_input_for_trainer module gigl.src.mocking.lib.mock_output_for_inference module gigl.src.mocking.lib.mocked_dataset_resources module gigl.src.mocking.lib.pyg_datasets_forks module gigl.src.mocking.lib.pyg_to_training_samples module gigl.src.mocking.lib.tfrecord_io module gigl.src.mocking.lib.user_defined_edge_sampling module gigl.src.mocking.lib.versioning module gigl.src.mocking.mocking_assets module gigl.src.mocking.mocking_assets.mocked_datasets_for_pipeline_tests module gigl.src.mocking.mocking_assets.passthrough_preprocessor_config_for_mocked_assets module gigl.src.mocking.toy_asset_mocker module gigl.src.post_process module gigl.src.post_process.lib module gigl.src.post_process.lib.base_post_processor module gigl.src.post_process.post_processor module gigl.src.post_process.utils module gigl.src.post_process.utils.component_runtime module gigl.src.post_process.utils.cosine_similarity module gigl.src.post_process.utils.unenumeration module gigl.src.split_generator module gigl.src.split_generator.split_generator module gigl.src.subgraph_sampler module gigl.src.subgraph_sampler.lib module gigl.src.subgraph_sampler.lib.ingestion_protocol module gigl.src.subgraph_sampler.subgraph_sampler module gigl.src.training module gigl.src.training.trainer module gigl.src.training.v1 module gigl.src.training.v1.lib module gigl.src.training.v1.lib.base_trainer module gigl.src.training.v1.lib.data_loaders module gigl.src.training.v1.lib.data_loaders.common module gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader module gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader module gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader module gigl.src.training.v1.lib.data_loaders.tf_records_iterable_dataset module gigl.src.training.v1.lib.data_loaders.utils module gigl.src.training.v1.lib.eval_metrics module gigl.src.training.v1.lib.training_process module gigl.src.training.v1.trainer module gigl.src.training.v2 module gigl.src.training.v2.glt_trainer module gigl.src.validation_check module gigl.src.validation_check.config_validator module gigl.src.validation_check.libs module gigl.src.validation_check.libs.frozen_config_path_checks module gigl.src.validation_check.libs.name_checks module gigl.src.validation_check.libs.resource_config_checks module gigl.src.validation_check.libs.template_config_checks module gigl.src.validation_check.libs.utils module gigl.types module gigl.types.graph module gigl.utils module gigl.utils.data_splitters module gigl.utils.share_memory module GIGL_DATAFLOW_IMAGE (in module gigl.env.dep_constants) GIGL_PUBLIC_BUCKET_NAME (in module gigl.env.dep_constants) GIGL_PUBLIC_DATASET_NAME (in module gigl.env.dep_constants) GIGL_ROOT_DIR (in module gigl.common.constants) gigl_root_dir (in module gigl.common.utils.jupyter_magics) GIGL_SRC_IMAGE_CPU (in module gigl.env.dep_constants) GIGL_SRC_IMAGE_CUDA (in module gigl.env.dep_constants) GIGL_TEST_BUCKET_NAME (in module gigl.env.dep_constants) GiGLComponents (class in gigl.src.common.constants.components) GiglResourceConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) GiglResourceConfigWrapper (class in gigl.src.common.types.pb_wrappers.gigl_resource_config) GIN (class in gigl.src.common.models.pyg.homogeneous) GINE (class in gigl.src.common.models.pyg.homogeneous) GINEConv (class in gigl.src.common.models.pyg.nn.conv.gin_conv) global___Component (in module snapchat.research.gbml.gigl_resource_config_pb2) global___DataflowResourceConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___DataPreprocessorConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___DatasetMetadata (in module snapchat.research.gbml.dataset_metadata_pb2) global___DistributedTrainerConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___Edge (in module snapchat.research.gbml.graph_schema_pb2) global___EdgeType (in module snapchat.research.gbml.graph_schema_pb2) global___FlattenedGraphMetadata (in module snapchat.research.gbml.flattened_graph_metadata_pb2) global___GbmlConfig (in module snapchat.research.gbml.gbml_config_pb2) global___GiglResourceConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___GlobalRandomUniformStrategy (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) global___Graph (in module snapchat.research.gbml.graph_schema_pb2) global___GraphMetadata (in module snapchat.research.gbml.graph_schema_pb2) global___InferenceMetadata (in module snapchat.research.gbml.inference_metadata_pb2) global___InferenceOutput (in module snapchat.research.gbml.inference_metadata_pb2) global___InferencerResourceConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___KFPResourceConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___KFPTrainerConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___Label (in module snapchat.research.gbml.training_samples_schema_pb2) global___LocalResourceConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___LocalTrainerConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___MessagePassingPath (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) global___MessagePassingPathStrategy (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) global___Node (in module snapchat.research.gbml.graph_schema_pb2) global___NodeAnchorBasedLinkPredictionDataset (in module snapchat.research.gbml.dataset_metadata_pb2) global___NodeAnchorBasedLinkPredictionOutput (in module snapchat.research.gbml.flattened_graph_metadata_pb2) global___NodeAnchorBasedLinkPredictionSample (in module snapchat.research.gbml.training_samples_schema_pb2) global___PostProcessedMetadata (in module snapchat.research.gbml.postprocessed_metadata_pb2) global___PreprocessedMetadata (in module snapchat.research.gbml.preprocessed_metadata_pb2) global___RandomUniform (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) global___RandomWeighted (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) global___RootedNodeNeighborhood (in module snapchat.research.gbml.training_samples_schema_pb2) global___SamplingDirection (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) global___SamplingOp (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) global___SharedResourceConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___SparkResourceConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___SubgraphSamplingStrategy (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) global___SupervisedLinkBasedTaskOutput (in module snapchat.research.gbml.flattened_graph_metadata_pb2) global___SupervisedLinkBasedTaskSample (in module snapchat.research.gbml.training_samples_schema_pb2) global___SupervisedLinkBasedTaskSplitDataset (in module snapchat.research.gbml.dataset_metadata_pb2) global___SupervisedNodeClassificationDataset (in module snapchat.research.gbml.dataset_metadata_pb2) global___SupervisedNodeClassificationOutput (in module snapchat.research.gbml.flattened_graph_metadata_pb2) global___SupervisedNodeClassificationSample (in module snapchat.research.gbml.training_samples_schema_pb2) global___TopK (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) global___TrainedModelMetadata (in module snapchat.research.gbml.trained_model_metadata_pb2) global___TrainerResourceConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___UserDefined (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) global___VertexAiResourceConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global___VertexAiTrainerConfig (in module snapchat.research.gbml.gigl_resource_config_pb2) global_node_to_subgraph_node_mapping (gigl.src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol property) (gigl.src.common.graph_builder.pyg_graph_data.PygGraphData property) global_random_uniform (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SubgraphSamplingStrategy property) GLOBAL_RANDOM_UNIFORM_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SubgraphSamplingStrategy attribute) global_random_uniform_strategy_pb (gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy.GlobalRandomUniformStrategyPbWrapper attribute) global_rank (gigl.distributed.dist_context.DistributedContext attribute) (gigl.distributed.DistributedContext attribute) global_world_size (gigl.distributed.dist_context.DistributedContext attribute) (gigl.distributed.DistributedContext attribute) GlobalRandomUniformStrategy (class in snapchat.research.gbml.subgraph_sampling_strategy_pb2) GlobalRandomUniformStrategyPbWrapper (class in gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy) GLT_BACKEND_UNSUPPORTED_COMPONENTS (in module gigl.src.common.constants.components) GLTInferencer (class in gigl.src.inference.v2.glt_inferencer) GLTTrainer (class in gigl.src.training.v2.glt_trainer) gnn_model (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec attribute) GnnModel (class in gigl.src.common.types.model) GnnTrainingProcess (class in gigl.src.training.v1.lib.training_process) gpu_limit (snapchat.research.gbml.gigl_resource_config_pb2.KFPResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.KFPTrainerConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiTrainerConfig attribute) GPU_LIMIT_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.KFPResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.KFPTrainerConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiTrainerConfig attribute) gpu_type (snapchat.research.gbml.gigl_resource_config_pb2.KFPResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.KFPTrainerConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiTrainerConfig attribute) GPU_TYPE_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.KFPResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.KFPTrainerConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiTrainerConfig attribute) GRACE (class in gigl.src.common.models.layers.task) GRACELoss (class in gigl.src.common.models.layers.loss) Graph (class in snapchat.research.gbml.graph_schema_pb2) graph (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) (gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader.NodeAnchorBasedLinkPredictionBatch attribute) (gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader.RootedNodeNeighborhoodBatch attribute) (gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader.SupervisedNodeClassificationBatch attribute) graph_backend (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN property) (gigl.src.common.models.pyg.homogeneous.TwoLayerGCN property) (gigl.src.common.models.pyg.link_prediction.LinkPredictionGNN property) (gigl.src.common.types.model.GnnModel property) graph_data_from_GraphPb() (gigl.src.common.translators.gbml_protos_translator.GbmlProtosTranslator static method) graph_db_args (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig property) GRAPH_DB_ARGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig attribute) graph_db_client_class_path (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDBServiceConfig attribute) GRAPH_DB_CLIENT_CLASS_PATH_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDBServiceConfig attribute) graph_db_config (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig property) GRAPH_DB_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) graph_db_ingestion_args (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig property) GRAPH_DB_INGESTION_ARGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig attribute) graph_db_ingestion_cls_path (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig attribute) GRAPH_DB_INGESTION_CLS_PATH_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig attribute) graph_db_sampler_config (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig property) GRAPH_DB_SAMPLER_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig attribute) graph_metadata (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig property) GRAPH_METADATA_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig attribute) graph_metadata_pb (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper attribute) graph_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo property) GraphBackend (class in gigl.src.common.types.model) GraphBuilder (class in gigl.src.common.graph_builder.abstract_graph_builder) GraphBuilderFactory (class in gigl.src.common.graph_builder.graph_builder_factory) GraphMetadata (class in snapchat.research.gbml.graph_schema_pb2) GraphMetadata.CondensedEdgeTypeMapEntry (class in snapchat.research.gbml.graph_schema_pb2) GraphMetadata.CondensedNodeTypeMapEntry (class in snapchat.research.gbml.graph_schema_pb2) GraphMetadataPbWrapper (class in gigl.src.common.types.pb_wrappers.graph_metadata) GraphPartitionData (class in gigl.types.graph) GraphPbWrapper (class in gigl.src.common.types.pb_wrappers.graph_data_types) GraphPbWrapperCoder (class in gigl.common.utils.compute.serialization.serialize_protos) GraphSAGE (class in gigl.src.common.models.pyg.homogeneous) GraphSageTemplateTrainerSpec (class in gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec) H hadamard_MLP (gigl.src.common.models.layers.decoder.DecoderType attribute) hard_neg_edge_features (gigl.src.common.translators.training_samples_protos_translator.NodeAnchorBasedLinkPredictionSample.SampleSupervisionEdgeData attribute) hard_neg_edges (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample property) HARD_NEG_EDGES_FIELD_NUMBER (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample attribute) hard_neg_embeddings (gigl.src.common.types.task_inputs.BatchEmbeddings attribute) hard_neg_ids (gigl.src.common.types.task_inputs.BatchCombinedScores attribute) hard_neg_nodes (gigl.src.common.translators.training_samples_protos_translator.NodeAnchorBasedLinkPredictionSample.SampleSupervisionEdgeData attribute) hard_neg_scores (gigl.src.common.types.task_inputs.BatchScores attribute) hard_neg_supervision_edge_data (gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader.NodeAnchorBasedLinkPredictionBatch attribute) has_hard_neg_edge_features() (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper method) has_pos_edge_features() (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper method) HasField() (snapchat.research.gbml.dataset_metadata_pb2.DatasetMetadata method) (snapchat.research.gbml.flattened_graph_metadata_pb2.FlattenedGraphMetadata method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.DataPreprocessorConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.DistributedTrainerConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.InferencerResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.TrainerResourceConfig method) (snapchat.research.gbml.graph_schema_pb2.Edge method) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedEdgeTypeMapEntry method) (snapchat.research.gbml.graph_schema_pb2.Node method) (snapchat.research.gbml.inference_metadata_pb2.InferenceMetadata.NodeTypeToInferencerOutputInfoMapEntry method) (snapchat.research.gbml.inference_metadata_pb2.InferenceOutput method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedEdgeTypeToPreprocessedMetadataEntry method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedNodeTypeToPreprocessedMetadataEntry method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.GlobalRandomUniformStrategy method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SubgraphSamplingStrategy method) (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample method) (snapchat.research.gbml.training_samples_schema_pb2.RootedNodeNeighborhood method) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedLinkBasedTaskSample method) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedNodeClassificationSample method) HashedNodeAnchorLinkSplitter (class in gigl.utils.data_splitters) head (gigl.src.common.models.layers.task.BGRL attribute) (gigl.src.common.models.layers.task.GRACE attribute) (gigl.src.common.models.layers.task.TBGRL attribute) (gigl.src.common.models.layers.task.WhiteningDecorrelation attribute) heads (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) HETEROGENEOUS_TOY_GRAPH_NODE_ANCHOR_MOCKED_DATASET_INFO (in module gigl.src.mocking.mocking_assets.mocked_datasets_for_pipeline_tests) HGT (class in gigl.src.common.models.pyg.heterogeneous) HGTConv (class in gigl.src.common.models.pyg.nn.conv.hgt_conv) hid_dim (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) (gigl.src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge attribute) hidden_dim (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec attribute) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec attribute) hit_rate_at_k() (in module gigl.src.common.utils.eval_metrics) HIT_RATE_METRIC_PREFIX (in module gigl.src.training.v1.lib.eval_metrics) hits (gigl.src.common.types.model_eval_metrics.EvalMetricType attribute) homogeneous_condensed_edge_type (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) homogeneous_condensed_node_type (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) homogeneous_edge_type (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) homogeneous_node_type (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) HttpUri (class in gigl.common) (class in gigl.common.types.uri.http_uri) hydrate() (gigl.src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper method) (gigl.src.common.types.pb_wrappers.graph_data_types.NodePbWrapper method) hydrate_subgraph_features() (gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper static method) I id (gigl.src.common.types.graph_data.Node attribute) identifier (gigl.src.data_preprocessor.lib.ingest.reference.NodeDataReference attribute) identifier_output (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) (gigl.src.data_preprocessor.lib.types.EdgeDataPreprocessingSpec attribute) (gigl.src.data_preprocessor.lib.types.NodeDataPreprocessingSpec attribute) IDLE_TTL_DEFAULT_S (in module gigl.src.common.utils.spark_job_manager) IDLE_TTL_DEV_DEFAULT_S (in module gigl.src.common.utils.spark_job_manager) ids (gigl.types.graph.FeaturePartitionData attribute) import_obj() (in module gigl.common.utils.os_utils) in_channels (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) in_dim (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) (gigl.src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction attribute) INCOMING (gigl.src.common.types.pb_wrappers.sampling_op.SamplingDirectionEnumType attribute) (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) IncrementalTaskIdentifier (in module gigl.src.common.types) infer_batch() (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec method) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec method) (gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec.NodeClassificationModelingTaskSpec method) (gigl.src.inference.v1.lib.base_inferencer.BaseInferencer method) infer_model() (in module gigl.src.mocking.lib.mock_output_for_inference) infer_root_embeddings() (in module gigl.src.common.modeling_task_specs.utils.infer) infer_task_inputs() (in module gigl.src.common.modeling_task_specs.utils.infer) infer_training_batch() (in module gigl.src.common.modeling_task_specs.utils.infer) InferBatchResults (class in gigl.src.inference.v1.lib.base_inferencer) inference_batch_size (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig attribute) INFERENCE_BATCH_SIZE_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig attribute) inference_metadata (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig property) INFERENCE_METADATA_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) InferenceAssets (class in gigl.src.inference.lib.assets) InferenceBatchType (in module gigl.src.inference.v1.lib.transforms.batch_generator) InferenceBlueprintFactory (class in gigl.src.inference.v1.lib.inference_blueprint_factory) InferenceMetadata (class in snapchat.research.gbml.inference_metadata_pb2) InferenceMetadata.NodeTypeToInferencerOutputInfoMapEntry (class in snapchat.research.gbml.inference_metadata_pb2) InferenceOutput (class in snapchat.research.gbml.inference_metadata_pb2) InferenceOutputBigqueryTableSchema (class in gigl.src.inference.v1.lib.inference_output_schema) InferenceOutputBigqueryTableSchemaBuilder (class in gigl.src.inference.v1.lib.inference_output_schema) Inferencer (class in gigl.src.inference.inferencer) (gigl.src.common.constants.components.GiGLComponents attribute) inferencer_args (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig property) INFERENCER_ARGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig attribute) inferencer_cls_path (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig attribute) INFERENCER_CLS_PATH_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig attribute) inferencer_config (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig property) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig property) INFERENCER_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig attribute) inferencer_resource_config (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig property) INFERENCER_RESOURCE_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig attribute) InferencerOutputPaths (class in gigl.src.inference.v1.gnn_inferencer) InferencerResourceConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) InferencerV1 (class in gigl.src.inference.v1.gnn_inferencer) ingest() (gigl.src.subgraph_sampler.lib.ingestion_protocol.BaseIngestion method) IngestRawFeatures (class in gigl.src.data_preprocessor.lib.transform.utils) init_beam_pipeline_options() (in module gigl.src.common.utils.dataflow) init_conv_layers() (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN method) (gigl.src.common.models.pyg.homogeneous.EdgeAttrGAT method) (gigl.src.common.models.pyg.homogeneous.GAT method) (gigl.src.common.models.pyg.homogeneous.GATv2 method) (gigl.src.common.models.pyg.homogeneous.GIN method) (gigl.src.common.models.pyg.homogeneous.GINE method) (gigl.src.common.models.pyg.homogeneous.GraphSAGE method) (gigl.src.common.models.pyg.homogeneous.Transformer method) init_metrics_publisher_grouping_for_job() (in module gigl.src.common.utils.metrics_service_provider) init_model() (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec method) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec method) (gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec.NodeClassificationModelingTaskSpec method) (gigl.src.common.types.model.BaseModelOperationsProtocol method) init_neighbor_loader_worker() (in module gigl.distributed.utils) (in module gigl.distributed.utils.init_neighbor_loader_worker) init_script_uri (gigl.src.common.utils.spark_job_manager.DataprocClusterInitData attribute) initial_eps (gigl.src.common.models.pyg.nn.conv.gin_conv.GINEConv attribute) initialize_metrics() (in module gigl.src.common.utils.metrics_service_provider) inner_product (gigl.src.common.models.layers.decoder.DecoderType attribute) input_batch (gigl.src.common.models.layers.loss.ModelResultType attribute) (gigl.src.common.types.task_inputs.NodeAnchorBasedLinkPredictionTaskInputs attribute) input_edge_data_reference (gigl.src.data_preprocessor.lib.enumerate.utils.EnumeratorEdgeTypeMetadata attribute) input_node_data_reference (gigl.src.data_preprocessor.lib.enumerate.utils.EnumeratorNodeTypeMetadata attribute) input_op_names (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper property) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp property) INPUT_OP_NAMES_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp attribute) InputBatch (class in gigl.src.common.types.task_inputs) InstanceDict (in module gigl.src.data_preprocessor.lib.types) InstanceDictPTransform (class in gigl.src.data_preprocessor.lib.types) InstanceDictToTFExample (class in gigl.src.data_preprocessor.lib.transform.utils) is_currently_running_in_vertex_ai_job() (in module gigl.common.utils.vertex_ai_context) is_datetime_str_format_valid() (in module gigl.src.common.utils.time) is_debug_mode (gigl.src.common.utils.spark_job_manager.DataprocClusterInitData attribute) is_deterministic() (gigl.common.beam.coders.PassthroughCoder method) (gigl.common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder method) (gigl.common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder method) (gigl.common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder method) is_distributed_available_and_initialized() (in module gigl.common.utils.torch_training) is_distributed_local_debug() (in module gigl.common.utils.torch_training) is_graph_directed (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) IS_GRAPH_DIRECTED_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) is_heterogeneous (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) is_node_entity (gigl.common.data.dataloaders.SerializedTFRecordInfo property) is_resource_config_loaded() (in module gigl.env.pipelines_config) is_training (gigl.src.common.models.pyg.homogeneous.TwoLayerGCN attribute) is_valid() (gigl.common.GcsUri class method) (gigl.common.HttpUri class method) (gigl.common.LocalUri class method) (gigl.common.types.uri.gcs_uri.GcsUri class method) (gigl.common.types.uri.http_uri.HttpUri class method) (gigl.common.types.uri.local_uri.LocalUri class method) (gigl.common.types.uri.uri.Uri class method) (gigl.common.Uri class method) J job_client (gigl.common.services.dataproc.DataprocService attribute) job_name (gigl.common.services.vertex_ai.VertexAiJobConfig attribute) JOB_NAME_GROUPING_ENV_KEY (in module gigl.src.common.utils.metrics_service_provider) job_parameters (gigl.common.types.wrappers.kfp_api.ApiRunDetailWrapper property) join() (gigl.common.GcsUri class method) (gigl.common.HttpUri class method) (gigl.common.LocalUri class method) (gigl.common.types.uri.gcs_uri.GcsUri class method) (gigl.common.types.uri.http_uri.HttpUri class method) (gigl.common.types.uri.local_uri.LocalUri class method) (gigl.common.types.uri.uri.Uri class method) (gigl.common.Uri class method) join_path() (gigl.src.common.utils.bq.BqUtils static method) json_serializable (gigl.src.mocking.lib.versioning.MockedDatasetArtifactMetadata property) JumpingKnowledge (class in gigl.src.common.models.pyg.nn.models.jumping_knowledge) K k_rel (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) kebab_case_value (gigl.src.common.constants.components.GiGLComponents property) key (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.TestNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.TrainNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.ValNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.NodeAnchorBasedLinkPredictionOutput.NodeTypeToRandomNegativeTfrecordUriPrefixEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.FeatureFlagsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDbArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig.InferencerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.MetricsConfig.MetricsArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig.ProfilerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig.SharedArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig.TrainerArgsEntry attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.ResourceLabelsEntry attribute) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedEdgeTypeMapEntry attribute) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedNodeTypeMapEntry attribute) (snapchat.research.gbml.inference_metadata_pb2.InferenceMetadata.NodeTypeToInferencerOutputInfoMapEntry attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedEdgeTypeToPreprocessedMetadataEntry attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedNodeTypeToPreprocessedMetadataEntry attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.UserDefined.ParamsEntry attribute) KEY_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.TestNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.TrainNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.ValNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.NodeAnchorBasedLinkPredictionOutput.NodeTypeToRandomNegativeTfrecordUriPrefixEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.FeatureFlagsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDbArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig.InferencerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.MetricsConfig.MetricsArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig.ProfilerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig.SharedArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig.TrainerArgsEntry attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.ResourceLabelsEntry attribute) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedEdgeTypeMapEntry attribute) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedNodeTypeMapEntry attribute) (snapchat.research.gbml.inference_metadata_pb2.InferenceMetadata.NodeTypeToInferencerOutputInfoMapEntry attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedEdgeTypeToPreprocessedMetadataEntry attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedNodeTypeToPreprocessedMetadataEntry attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.UserDefined.ParamsEntry attribute) kfp_trainer_config (snapchat.research.gbml.gigl_resource_config_pb2.DistributedTrainerConfig property) (snapchat.research.gbml.gigl_resource_config_pb2.TrainerResourceConfig property) KFP_TRAINER_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.DistributedTrainerConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.TrainerResourceConfig attribute) kfp_validation_checks() (in module gigl.src.validation_check.config_validator) KfpOrchestrator (class in gigl.orchestration.kubeflow.kfp_orchestrator) KfpOutputViewers (class in gigl.common.utils.kfp) KFPResourceConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) KfpTaskDetails (class in gigl.common.types.wrappers.kfp_api) KFPTrainerConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) kl_temperature (gigl.src.common.models.layers.loss.KLLoss attribute) KLLoss (class in gigl.src.common.models.layers.loss) kqv_lin (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) KS_FOR_EVAL (in module gigl.src.training.v1.lib.eval_metrics) KT (in module gigl.common.collections.frozen_dict) (in module gigl.common.collections.sorted_dict) L l2_normalize_embeddings() (in module gigl.src.common.models.layers.normalization) Label (class in snapchat.research.gbml.training_samples_schema_pb2) label (snapchat.research.gbml.training_samples_schema_pb2.Label attribute) label_edge_features (gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader.NodeAnchorBasedLinkPredictionBatch.BatchSupervisionEdgeData attribute) LABEL_FIELD_NUMBER (snapchat.research.gbml.training_samples_schema_pb2.Label attribute) label_keys (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo property) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput property) LABEL_KEYS_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) label_outputs (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) label_type (snapchat.research.gbml.training_samples_schema_pb2.Label attribute) LABEL_TYPE_FIELD_NUMBER (snapchat.research.gbml.training_samples_schema_pb2.Label attribute) labeled_tfrecord_uri_prefix (snapchat.research.gbml.flattened_graph_metadata_pb2.SupervisedLinkBasedTaskOutput attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.SupervisedNodeClassificationOutput attribute) LABELED_TFRECORD_URI_PREFIX_FIELD_NUMBER (snapchat.research.gbml.flattened_graph_metadata_pb2.SupervisedLinkBasedTaskOutput attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.SupervisedNodeClassificationOutput attribute) labels (gigl.common.services.vertex_ai.VertexAiJobConfig attribute) (gigl.src.common.utils.spark_job_manager.DataprocClusterInitData attribute) labels_outputs (gigl.src.data_preprocessor.lib.types.EdgeDataPreprocessingSpec attribute) (gigl.src.data_preprocessor.lib.types.NodeDataPreprocessingSpec attribute) lambd (gigl.src.common.models.layers.loss.WhiteningDecorrelationLoss attribute) launch_job() (gigl.common.services.vertex_ai.VertexAIService method) LEADER_WORKER_INTERNAL_IP_FILE_PATH_ENV_KEY (in module gigl.common.services.vertex_ai) leakyrelu (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) levels (gigl.common.metrics.base_metrics.NopMetricsPublisher attribute) lin (gigl.src.common.models.pyg.heterogeneous.HGT attribute) (gigl.src.common.models.pyg.heterogeneous.SimpleHGN attribute) lin_dict (gigl.src.common.models.pyg.heterogeneous.HGT attribute) linear_layer (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) LINK_BASED_TASK (gigl.src.common.types.task_metadata.TaskMetadataType attribute) link_based_task_metadata (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata property) LINK_BASED_TASK_METADATA_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata attribute) LINK_PREDICTION_METRICS_PREFIX (in module gigl.src.training.v1.lib.eval_metrics) LinkPredictionDecoder (class in gigl.src.common.models.layers.decoder) LinkPredictionGNN (class in gigl.src.common.models.pyg.link_prediction) list_at_path() (in module gigl.common.utils.local_fs) list_children() (gigl.src.common.utils.file_loader.FileLoader method) list_matching_tables() (gigl.src.common.utils.bq.BqUtils method) list_uris_with_gcs_path_pattern() (gigl.common.utils.gcs.GcsUtils method) LLPRankingLoss (class in gigl.src.common.models.layers.loss) load() (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset method) (gigl.distributed.DistLinkPredictionDataset method) load_as_torch_tensors() (gigl.common.data.dataloaders.TFRecordDataLoader method) load_directories() (gigl.src.common.utils.file_loader.FileLoader method) load_directory() (gigl.src.common.utils.file_loader.FileLoader method) load_embeddings_to_bigquery() (in module gigl.common.data.export) load_file() (gigl.src.common.utils.file_loader.FileLoader method) load_file_to_bq() (gigl.src.common.utils.bq.BqUtils method) load_files() (gigl.src.common.utils.file_loader.FileLoader method) load_preprocessed_metadata_pb_wrapper_from_gbml_config_pb() (in module gigl.src.mocking.lib.gbml_config) load_rows_to_bq() (gigl.src.common.utils.bq.BqUtils method) load_scripted_model_from_uri() (in module gigl.src.common.utils.model) load_state_dict_from_uri() (in module gigl.src.common.utils.model) load_tf_schema_uri_str_to_feature_spec() (in module gigl.common.utils.tensorflow_schema) load_to_temp_file() (gigl.src.common.utils.file_loader.FileLoader method) load_torch_tensors_from_tf_record() (in module gigl.common.data.load_torch_tensors) load_toy_graph() (in module gigl.src.mocking.toy_asset_mocker) LoadedGraphTensors (class in gigl.types.graph) local_inferencer_config (snapchat.research.gbml.gigl_resource_config_pb2.InferencerResourceConfig property) LOCAL_INFERENCER_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.InferencerResourceConfig attribute) local_trainer_config (snapchat.research.gbml.gigl_resource_config_pb2.DistributedTrainerConfig property) (snapchat.research.gbml.gigl_resource_config_pb2.TrainerResourceConfig property) LOCAL_TRAINER_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.DistributedTrainerConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.TrainerResourceConfig attribute) LocalResourceConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) LocalTrainerConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) LocalUri (class in gigl.common) (class in gigl.common.types.uri.local_uri) log_metrics_to_ui() (in module gigl.orchestration.kubeflow.utils.log_metrics) log_stats_for_pyg_planetoid_dataset() (in module gigl.src.applied_tasks.test_tasks.academic) LogEntries (class in gigl.common.services.gcp_cloud_logging) Logger (class in gigl.common.logger) logger (in module gigl.analytics.graph_validation.bq_graph_validator) (in module gigl.common.beam.better_tfrecordio) (in module gigl.common.data.dataloaders) (in module gigl.common.data.export) (in module gigl.common.data.load_torch_tensors) (in module gigl.common.metrics.decorators) (in module gigl.common.services.dataproc) (in module gigl.common.services.gcp_cloud_logging) (in module gigl.common.services.vertex_ai) (in module gigl.common.types.wrappers.kfp_api) (in module gigl.common.utils.compute.random) (in module gigl.common.utils.dataflow) (in module gigl.common.utils.gcs) (in module gigl.common.utils.kfp) (in module gigl.common.utils.local_fs) (in module gigl.common.utils.os_utils) (in module gigl.common.utils.proto_utils) (in module gigl.common.utils.retry) (in module gigl.common.utils.test_utils) (in module gigl.common.utils.torch_training) (in module gigl.common.utils.vertex_ai_context) (in module gigl.distributed.dataset_factory) (in module gigl.distributed.dist_link_prediction_dataset) (in module gigl.distributed.dist_partitioner) (in module gigl.distributed.dist_range_partitioner) (in module gigl.distributed.distributed_neighborloader) (in module gigl.distributed.utils.init_neighbor_loader_worker) (in module gigl.env.pipelines_config) (in module gigl.orchestration.kubeflow.kfp_orchestrator) (in module gigl.orchestration.kubeflow.kfp_pipeline) (in module gigl.orchestration.kubeflow.runner) (in module gigl.orchestration.local.runner) (in module gigl.src.applied_tasks.test_tasks.academic) (in module gigl.src.common.graph_builder.pyg_graph_builder) (in module gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec) (in module gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec) (in module gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec) (in module gigl.src.common.modeling_task_specs.utils.early_stop) (in module gigl.src.common.modeling_task_specs.utils.profiler_wrapper) (in module gigl.src.common.models.layers.count_min_sketch) (in module gigl.src.common.models.layers.task) (in module gigl.src.common.models.pyg.homogeneous) (in module gigl.src.common.models.pyg.nn.models.feature_embedding) (in module gigl.src.common.translators.gbml_protos_translator) (in module gigl.src.common.translators.training_samples_protos_translator) (in module gigl.src.common.types.pb_wrappers.dataset_metadata_utils) (in module gigl.src.common.types.pb_wrappers.gbml_config) (in module gigl.src.common.types.pb_wrappers.gigl_resource_config) (in module gigl.src.common.types.pb_wrappers.graph_metadata) (in module gigl.src.common.types.pb_wrappers.preprocessed_metadata) (in module gigl.src.common.utils.bq) (in module gigl.src.common.utils.dataflow) (in module gigl.src.common.utils.file_loader) (in module gigl.src.common.utils.metrics_service_provider) (in module gigl.src.common.utils.spark_job_manager) (in module gigl.src.common.utils.timeout) (in module gigl.src.config_populator.config_populator) (in module gigl.src.data_preprocessor.data_preprocessor) (in module gigl.src.data_preprocessor.lib.data_preprocessor_config) (in module gigl.src.data_preprocessor.lib.enumerate.utils) (in module gigl.src.data_preprocessor.lib.transform.utils) (in module gigl.src.inference.inferencer) (in module gigl.src.inference.lib.assets) (in module gigl.src.inference.v1.gnn_inferencer) (in module gigl.src.inference.v1.lib.base_inferencer) (in module gigl.src.inference.v1.lib.utils) (in module gigl.src.inference.v2.glt_inferencer) (in module gigl.src.mocking.dataset_asset_mocker) (in module gigl.src.mocking.dataset_asset_mocking_suite) (in module gigl.src.mocking.lib.mock_input_for_data_preprocessor) (in module gigl.src.mocking.lib.mock_input_for_inference) (in module gigl.src.mocking.lib.mock_input_for_split_generator) (in module gigl.src.mocking.lib.mock_input_for_subgraph_sampler) (in module gigl.src.mocking.lib.mock_input_for_trainer) (in module gigl.src.mocking.lib.mock_output_for_inference) (in module gigl.src.mocking.lib.tfrecord_io) (in module gigl.src.mocking.lib.versioning) (in module gigl.src.post_process.lib.base_post_processor) (in module gigl.src.post_process.post_processor) (in module gigl.src.post_process.utils.unenumeration) (in module gigl.src.split_generator.split_generator) (in module gigl.src.subgraph_sampler.subgraph_sampler) (in module gigl.src.training.trainer) (in module gigl.src.training.v1.lib.base_trainer) (in module gigl.src.training.v1.lib.data_loaders.common) (in module gigl.src.training.v1.lib.data_loaders.utils) (in module gigl.src.training.v1.lib.training_process) (in module gigl.src.training.v1.trainer) (in module gigl.src.training.v2.glt_trainer) (in module gigl.src.validation_check.config_validator) (in module gigl.src.validation_check.libs.frozen_config_path_checks) (in module gigl.src.validation_check.libs.name_checks) (in module gigl.src.validation_check.libs.resource_config_checks) (in module gigl.src.validation_check.libs.template_config_checks) (in module gigl.types.graph) (in module gigl.utils.data_splitters) LoopyIterableDataset (class in gigl.src.training.v1.lib.data_loaders.tf_records_iterable_dataset) loss (gigl.src.common.models.layers.task.BGRL attribute) (gigl.src.common.models.layers.task.FeatureReconstruction attribute) (gigl.src.common.models.layers.task.GBT attribute) (gigl.src.common.models.layers.task.GRACE attribute) (gigl.src.common.models.layers.task.Margin attribute) (gigl.src.common.models.layers.task.Retrieval attribute) (gigl.src.common.models.layers.task.Softmax attribute) (gigl.src.common.models.layers.task.TBGRL attribute) (gigl.src.common.models.layers.task.WhiteningDecorrelation attribute) (gigl.src.common.types.model_eval_metrics.EvalMetricType attribute) LOSS_METRIC_NAME (in module gigl.src.training.v1.lib.eval_metrics) lru_cache() (in module gigl.common.utils.func_tools) M machine_type (gigl.common.services.vertex_ai.VertexAiJobConfig attribute) (gigl.src.common.utils.spark_job_manager.DataprocClusterInitData attribute) (snapchat.research.gbml.gigl_resource_config_pb2.DataflowResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SparkResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiTrainerConfig attribute) MACHINE_TYPE_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.DataflowResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SparkResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiTrainerConfig attribute) MAIN (gigl.src.common.types.graph_data.EdgeUsageType attribute) main_batch (gigl.src.common.types.task_inputs.InputBatch attribute) main_edge_info (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput property) MAIN_EDGE_INFO_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput attribute) main_sample_batch_size (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec attribute) main_worker_ip_address (gigl.distributed.dist_context.DistributedContext attribute) (gigl.distributed.DistributedContext attribute) make_compute_deterministic_and_set_seed() (in module gigl.common.utils.compute.random) Margin (class in gigl.src.common.models.layers.task) margin (gigl.src.common.models.layers.loss.LLPRankingLoss attribute) (gigl.src.common.models.layers.loss.MarginLoss attribute) margin_loss (gigl.src.common.models.layers.loss.LLPRankingLoss attribute) MarginLoss (class in gigl.src.common.models.layers.loss) MAX_INFERENCER_NUM_WORKERS (in module gigl.src.inference.v1.gnn_inferencer) MAX_JOB_DURATION (in module gigl.src.split_generator.split_generator) (in module gigl.src.subgraph_sampler.subgraph_sampler) max_num_workers (snapchat.research.gbml.gigl_resource_config_pb2.DataflowResourceConfig attribute) MAX_NUM_WORKERS_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.DataflowResourceConfig attribute) max_val (gigl.src.common.utils.scheduler.CosineDecayScheduler attribute) mean_reciprocal_rank() (in module gigl.src.common.utils.eval_metrics) memory_request (snapchat.research.gbml.gigl_resource_config_pb2.KFPResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.KFPTrainerConfig attribute) MEMORY_REQUEST_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.KFPResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.KFPTrainerConfig attribute) merge_subgraphs() (gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper class method) message (gigl.src.common.types.exception.SubgraphSamplingValidationError attribute) message() (gigl.src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (gigl.src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) MESSAGE_PASSING_BASE_CLS_ARGS (in module gigl.src.common.models.pyg.utils) message_passing_path_pb (gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathPbWrapper attribute) message_passing_path_strategy_pb (gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathStrategyPbWrapper attribute) message_passing_paths (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SubgraphSamplingStrategy property) MESSAGE_PASSING_PATHS_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SubgraphSamplingStrategy attribute) message_passing_to_negative_label() (in module gigl.types.graph) message_passing_to_positive_label() (in module gigl.types.graph) MessagePassingPath (class in snapchat.research.gbml.subgraph_sampling_strategy_pb2) MessagePassingPathPbWrapper (class in gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy) MessagePassingPathStrategy (class in snapchat.research.gbml.subgraph_sampling_strategy_pb2) MessagePassingPathStrategyPbWrapper (class in gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy) metrics (gigl.src.common.types.model_eval_metrics.EvalMetricsCollection property) metrics_args (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.MetricsConfig property) METRICS_ARGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.MetricsConfig attribute) metrics_cls_path (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.MetricsConfig attribute) METRICS_CLS_PATH_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.MetricsConfig attribute) metrics_config (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig property) METRICS_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig attribute) MILLISECONDS (gigl.common.metrics.decorators.TimerRecordGranularity attribute) MISSING_EXPECTED_ROOT_NODE_TYPE (gigl.src.common.types.exception.SubgraphSamplingValidationErrorType attribute) MISSING_ROOT_SAMPLING_OP (gigl.src.common.types.exception.SubgraphSamplingValidationErrorType attribute) mlp (gigl.src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction attribute) mock_assets() (gigl.src.mocking.dataset_asset_mocker.DatasetAssetMocker method) mock_cora_homogeneous_node_anchor_based_link_prediction_dataset() (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite method) mock_cora_homogeneous_node_anchor_based_link_prediction_dataset_with_edge_features() (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite method) mock_cora_homogeneous_node_anchor_based_link_prediction_dataset_with_user_defined_labels() (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite method) mock_cora_homogeneous_supervised_node_classification_dataset() (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite method) mock_cora_homogeneous_supervised_node_classification_dataset_with_edge_features() (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite method) MOCK_DATA_BQ_DATASET_NAME (in module gigl.src.mocking.lib.constants) MOCK_DATA_GCS_BUCKET (in module gigl.src.mocking.lib.constants) mock_dblp_node_anchor_based_link_prediction_dataset() (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite method) mock_dblp_node_anchor_based_link_prediction_dataset_with_user_defined_labels() (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite method) mock_toy_graph_heterogeneous_node_anchor_based_link_prediction_dataset() (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite method) mock_toy_graph_homogeneous_node_anchor_based_link_prediction_dataset() (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite method) mock_toy_graph_homogeneous_node_anchor_based_link_prediction_with_user_def_labels_dataset() (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite method) MOCKED_DATASET_ARTIFACT_METADATA_LOCAL_PATH (in module gigl.src.mocking.lib.constants) MockedDatasetArtifactMetadata (class in gigl.src.mocking.lib.versioning) MockedDatasetInfo (class in gigl.src.mocking.lib.mocked_dataset_resources) mode (gigl.src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge attribute) model (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec property) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec property) (gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec.NodeClassificationModelingTaskSpec property) (gigl.src.common.types.model.BaseModelOperationsProtocol property) ModelResultType (class in gigl.src.common.models.layers.loss) module gigl gigl.analytics gigl.analytics.graph_validation gigl.analytics.graph_validation.bq_graph_validator gigl.common gigl.common.beam gigl.common.beam.better_tfrecordio gigl.common.beam.coders gigl.common.collections gigl.common.collections.frozen_dict gigl.common.collections.itertools gigl.common.collections.sorted_dict gigl.common.constants gigl.common.data gigl.common.data.dataloaders gigl.common.data.export gigl.common.data.load_torch_tensors gigl.common.env_config gigl.common.logger gigl.common.metrics gigl.common.metrics.base_metrics gigl.common.metrics.decorators gigl.common.metrics.metrics_interface gigl.common.services gigl.common.services.dataproc gigl.common.services.gcp_cloud_logging gigl.common.services.vertex_ai gigl.common.types gigl.common.types.resource_config gigl.common.types.uri gigl.common.types.uri.gcs_uri gigl.common.types.uri.http_uri gigl.common.types.uri.local_uri gigl.common.types.uri.uri gigl.common.types.uri.uri_factory gigl.common.types.wrappers gigl.common.types.wrappers.kfp_api gigl.common.utils gigl.common.utils.compute gigl.common.utils.compute.random gigl.common.utils.compute.serialization gigl.common.utils.compute.serialization.coder gigl.common.utils.compute.serialization.serialize_np gigl.common.utils.compute.serialization.serialize_protos gigl.common.utils.compute.serialization.serialize_string gigl.common.utils.dataflow gigl.common.utils.decorator gigl.common.utils.func_tools gigl.common.utils.gcs gigl.common.utils.jupyter_magics gigl.common.utils.kfp gigl.common.utils.local_fs gigl.common.utils.os_utils gigl.common.utils.proto_utils gigl.common.utils.retry gigl.common.utils.tensorflow_schema gigl.common.utils.test_utils gigl.common.utils.torch_training gigl.common.utils.vertex_ai_context gigl.distributed gigl.distributed.constants gigl.distributed.dataset_factory gigl.distributed.dist_context gigl.distributed.dist_link_prediction_dataset gigl.distributed.dist_partitioner gigl.distributed.dist_range_partitioner gigl.distributed.distributed_neighborloader gigl.distributed.utils gigl.distributed.utils.device gigl.distributed.utils.init_neighbor_loader_worker gigl.distributed.utils.partition_book gigl.distributed.utils.serialized_graph_metadata_translator gigl.env gigl.env.dep_constants gigl.env.pipelines_config gigl.orchestration gigl.orchestration.kubeflow gigl.orchestration.kubeflow.kfp_orchestrator gigl.orchestration.kubeflow.kfp_pipeline gigl.orchestration.kubeflow.runner gigl.orchestration.kubeflow.utils gigl.orchestration.kubeflow.utils.glt_backend gigl.orchestration.kubeflow.utils.log_metrics gigl.orchestration.kubeflow.utils.resource gigl.orchestration.local gigl.orchestration.local.runner gigl.src gigl.src.applied_tasks gigl.src.applied_tasks.test_tasks gigl.src.applied_tasks.test_tasks.academic gigl.src.common gigl.src.common.constants gigl.src.common.constants.bq gigl.src.common.constants.components gigl.src.common.constants.gcs gigl.src.common.constants.graph_metadata gigl.src.common.constants.local_fs gigl.src.common.constants.metrics gigl.src.common.constants.resource_config gigl.src.common.constants.time gigl.src.common.constants.training gigl.src.common.graph_builder gigl.src.common.graph_builder.abstract_graph_builder gigl.src.common.graph_builder.gbml_graph_protocol gigl.src.common.graph_builder.graph_builder_factory gigl.src.common.graph_builder.pyg_graph_builder gigl.src.common.graph_builder.pyg_graph_data gigl.src.common.modeling_task_specs gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec gigl.src.common.modeling_task_specs.utils gigl.src.common.modeling_task_specs.utils.early_stop gigl.src.common.modeling_task_specs.utils.infer gigl.src.common.modeling_task_specs.utils.parameter_loading gigl.src.common.modeling_task_specs.utils.profiler_wrapper gigl.src.common.models gigl.src.common.models.dgl gigl.src.common.models.dgl.homogenous gigl.src.common.models.layers gigl.src.common.models.layers.count_min_sketch gigl.src.common.models.layers.decoder gigl.src.common.models.layers.feature_interaction gigl.src.common.models.layers.loss gigl.src.common.models.layers.normalization gigl.src.common.models.layers.task gigl.src.common.models.pyg gigl.src.common.models.pyg.graph gigl.src.common.models.pyg.graph.augmentations gigl.src.common.models.pyg.heterogeneous gigl.src.common.models.pyg.homogeneous gigl.src.common.models.pyg.link_prediction gigl.src.common.models.pyg.nn gigl.src.common.models.pyg.nn.conv gigl.src.common.models.pyg.nn.conv.edge_attr_gat_conv gigl.src.common.models.pyg.nn.conv.gin_conv gigl.src.common.models.pyg.nn.conv.hgt_conv gigl.src.common.models.pyg.nn.conv.simplehgn_conv gigl.src.common.models.pyg.nn.models gigl.src.common.models.pyg.nn.models.feature_embedding gigl.src.common.models.pyg.nn.models.feature_interaction gigl.src.common.models.pyg.nn.models.jumping_knowledge gigl.src.common.models.pyg.utils gigl.src.common.models.utils gigl.src.common.models.utils.torch gigl.src.common.translators gigl.src.common.translators.gbml_protos_translator gigl.src.common.translators.model_eval_metrics_translator gigl.src.common.translators.training_samples_protos_translator gigl.src.common.types gigl.src.common.types.dataflow_job_options gigl.src.common.types.dataset_split gigl.src.common.types.exception gigl.src.common.types.features gigl.src.common.types.graph_data gigl.src.common.types.model gigl.src.common.types.model_eval_metrics gigl.src.common.types.pb_wrappers gigl.src.common.types.pb_wrappers.dataset_metadata gigl.src.common.types.pb_wrappers.dataset_metadata_utils gigl.src.common.types.pb_wrappers.flattened_graph_metadata gigl.src.common.types.pb_wrappers.gbml_config gigl.src.common.types.pb_wrappers.gigl_resource_config gigl.src.common.types.pb_wrappers.graph_data_types gigl.src.common.types.pb_wrappers.graph_data_types_utils gigl.src.common.types.pb_wrappers.graph_metadata gigl.src.common.types.pb_wrappers.preprocessed_metadata gigl.src.common.types.pb_wrappers.sampling_op gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy gigl.src.common.types.pb_wrappers.task_metadata gigl.src.common.types.pb_wrappers.trained_model_metadata gigl.src.common.types.pb_wrappers.types gigl.src.common.types.task_inputs gigl.src.common.types.task_metadata gigl.src.common.utils gigl.src.common.utils.bq gigl.src.common.utils.data gigl.src.common.utils.data.feature_serialization gigl.src.common.utils.data.training gigl.src.common.utils.dataflow gigl.src.common.utils.eval_metrics gigl.src.common.utils.file_loader gigl.src.common.utils.gbml_protos gigl.src.common.utils.metrics_service_provider gigl.src.common.utils.model gigl.src.common.utils.scheduler gigl.src.common.utils.spark_job_manager gigl.src.common.utils.time gigl.src.common.utils.timeout gigl.src.config_populator gigl.src.config_populator.config_populator gigl.src.data_preprocessor gigl.src.data_preprocessor.data_preprocessor gigl.src.data_preprocessor.lib gigl.src.data_preprocessor.lib.data_preprocessor_config gigl.src.data_preprocessor.lib.enumerate gigl.src.data_preprocessor.lib.enumerate.queries gigl.src.data_preprocessor.lib.enumerate.utils gigl.src.data_preprocessor.lib.ingest gigl.src.data_preprocessor.lib.ingest.bigquery gigl.src.data_preprocessor.lib.ingest.reference gigl.src.data_preprocessor.lib.transform gigl.src.data_preprocessor.lib.transform.tf_value_encoder gigl.src.data_preprocessor.lib.transform.transformed_features_info gigl.src.data_preprocessor.lib.transform.utils gigl.src.data_preprocessor.lib.types gigl.src.inference gigl.src.inference.inferencer gigl.src.inference.lib gigl.src.inference.lib.assets gigl.src.inference.v1 gigl.src.inference.v1.gnn_inferencer gigl.src.inference.v1.lib gigl.src.inference.v1.lib.base_inference_blueprint gigl.src.inference.v1.lib.base_inferencer gigl.src.inference.v1.lib.inference_blueprint_factory gigl.src.inference.v1.lib.inference_output_schema gigl.src.inference.v1.lib.node_anchor_based_link_prediction_inferencer gigl.src.inference.v1.lib.node_classification_inferencer gigl.src.inference.v1.lib.queries gigl.src.inference.v1.lib.transforms gigl.src.inference.v1.lib.transforms.batch_generator gigl.src.inference.v1.lib.transforms.utils gigl.src.inference.v1.lib.utils gigl.src.inference.v2 gigl.src.inference.v2.glt_inferencer gigl.src.mocking gigl.src.mocking.dataset_asset_mocker gigl.src.mocking.dataset_asset_mocking_suite gigl.src.mocking.lib gigl.src.mocking.lib.constants gigl.src.mocking.lib.feature_handling gigl.src.mocking.lib.gbml_config gigl.src.mocking.lib.mock_input_for_data_preprocessor gigl.src.mocking.lib.mock_input_for_inference gigl.src.mocking.lib.mock_input_for_split_generator gigl.src.mocking.lib.mock_input_for_subgraph_sampler gigl.src.mocking.lib.mock_input_for_trainer gigl.src.mocking.lib.mock_output_for_inference gigl.src.mocking.lib.mocked_dataset_resources gigl.src.mocking.lib.pyg_datasets_forks gigl.src.mocking.lib.pyg_to_training_samples gigl.src.mocking.lib.tfrecord_io gigl.src.mocking.lib.user_defined_edge_sampling gigl.src.mocking.lib.versioning gigl.src.mocking.mocking_assets gigl.src.mocking.mocking_assets.mocked_datasets_for_pipeline_tests gigl.src.mocking.mocking_assets.passthrough_preprocessor_config_for_mocked_assets gigl.src.mocking.toy_asset_mocker gigl.src.post_process gigl.src.post_process.lib gigl.src.post_process.lib.base_post_processor gigl.src.post_process.post_processor gigl.src.post_process.utils gigl.src.post_process.utils.component_runtime gigl.src.post_process.utils.cosine_similarity gigl.src.post_process.utils.unenumeration gigl.src.split_generator gigl.src.split_generator.split_generator gigl.src.subgraph_sampler gigl.src.subgraph_sampler.lib gigl.src.subgraph_sampler.lib.ingestion_protocol gigl.src.subgraph_sampler.subgraph_sampler gigl.src.training gigl.src.training.trainer gigl.src.training.v1 gigl.src.training.v1.lib gigl.src.training.v1.lib.base_trainer gigl.src.training.v1.lib.data_loaders gigl.src.training.v1.lib.data_loaders.common gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader gigl.src.training.v1.lib.data_loaders.tf_records_iterable_dataset gigl.src.training.v1.lib.data_loaders.utils gigl.src.training.v1.lib.eval_metrics gigl.src.training.v1.lib.training_process gigl.src.training.v1.trainer gigl.src.training.v2 gigl.src.training.v2.glt_trainer gigl.src.validation_check gigl.src.validation_check.config_validator gigl.src.validation_check.libs gigl.src.validation_check.libs.frozen_config_path_checks gigl.src.validation_check.libs.name_checks gigl.src.validation_check.libs.resource_config_checks gigl.src.validation_check.libs.template_config_checks gigl.src.validation_check.libs.utils gigl.types gigl.types.graph gigl.utils gigl.utils.data_splitters gigl.utils.share_memory snapchat snapchat.research snapchat.research.gbml snapchat.research.gbml.dataset_metadata_pb2 snapchat.research.gbml.flattened_graph_metadata_pb2 snapchat.research.gbml.gbml_config_pb2 snapchat.research.gbml.gigl_resource_config_pb2 snapchat.research.gbml.graph_schema_pb2 snapchat.research.gbml.inference_metadata_pb2 snapchat.research.gbml.postprocessed_metadata_pb2 snapchat.research.gbml.preprocessed_metadata_pb2 snapchat.research.gbml.subgraph_sampling_strategy_pb2 snapchat.research.gbml.trained_model_metadata_pb2 snapchat.research.gbml.training_samples_schema_pb2 mrr (gigl.src.common.types.model_eval_metrics.EvalMetricType attribute) MRR_METRIC_NAME (in module gigl.src.training.v1.lib.eval_metrics) N name (gigl.src.common.types.model_eval_metrics.EvalMetric attribute) (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) neg_edge_feats (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite.UserDefinedLabels attribute) neg_edge_index (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite.UserDefinedLabels attribute) neg_edges (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample property) NEG_EDGES_FIELD_NUMBER (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample attribute) neg_lambda (gigl.src.common.models.layers.loss.TBGRLLoss attribute) NEGATIVE (gigl.src.common.types.graph_data.EdgeUsageType attribute) negative_edge_info (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput property) NEGATIVE_EDGE_INFO_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput attribute) negative_edge_label (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) negative_label (gigl.types.graph.LoadedGraphTensors attribute) negative_label_entity_info (gigl.common.data.load_torch_tensors.SerializedGraphMetadata attribute) neighborhood (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample property) (snapchat.research.gbml.training_samples_schema_pb2.RootedNodeNeighborhood property) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedLinkBasedTaskSample property) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedNodeClassificationSample property) NEIGHBORHOOD_FIELD_NUMBER (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample attribute) (snapchat.research.gbml.training_samples_schema_pb2.RootedNodeNeighborhood attribute) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedLinkBasedTaskSample attribute) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedNodeClassificationSample attribute) nfeat_drop (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) nn (gigl.src.common.models.pyg.nn.conv.gin_conv.GINEConv attribute) NO_EDGE_FEATURES_GRAPH_EDGELIST_ENUMERATION_QUERY (in module gigl.src.data_preprocessor.lib.enumerate.queries) no_grad_eval() (in module gigl.src.inference.v1.lib.base_inferencer) NODASH_DATE_FMT (in module gigl.src.common.constants.time) NODASH_DATETIME_FORMAT (in module gigl.src.common.constants.time) Node (class in gigl.src.common.types.graph_data) (class in snapchat.research.gbml.graph_schema_pb2) NODE (gigl.src.common.types.features.FeatureTypes attribute) node_anchor_based_link_prediction_dataset (snapchat.research.gbml.dataset_metadata_pb2.DatasetMetadata property) NODE_ANCHOR_BASED_LINK_PREDICTION_DATASET_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.DatasetMetadata attribute) node_anchor_based_link_prediction_output (snapchat.research.gbml.flattened_graph_metadata_pb2.FlattenedGraphMetadata property) NODE_ANCHOR_BASED_LINK_PREDICTION_OUTPUT_FIELD_NUMBER (snapchat.research.gbml.flattened_graph_metadata_pb2.FlattenedGraphMetadata attribute) NODE_ANCHOR_BASED_LINK_PREDICTION_TASK (gigl.src.common.types.task_metadata.TaskMetadataType attribute) node_anchor_based_link_prediction_task_metadata (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata property) NODE_ANCHOR_BASED_LINK_PREDICTION_TASK_METADATA_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata attribute) NODE_BASED_TASK (gigl.src.common.types.task_metadata.TaskMetadataType attribute) node_based_task_metadata (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata property) NODE_BASED_TASK_METADATA_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata attribute) node_data (gigl.src.data_preprocessor.data_preprocessor.PreprocessedMetadataReferences attribute) node_entity_info (gigl.common.data.load_torch_tensors.SerializedGraphMetadata attribute) node_feats (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite.ToyGraphData attribute) (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) node_features (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) (gigl.types.graph.LoadedGraphTensors attribute) NODE_FEATURES_ENUMERATION_QUERY (in module gigl.src.data_preprocessor.lib.enumerate.queries) node_field (gigl.src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema attribute) node_from_NodePb() (gigl.src.common.translators.gbml_protos_translator.GbmlProtosTranslator static method) node_id (gigl.src.common.types.pb_wrappers.graph_data_types.NodePbWrapper property) (snapchat.research.gbml.graph_schema_pb2.Node attribute) node_id_column_name (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) NODE_ID_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.Node attribute) node_id_key (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) NODE_ID_KEY_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) node_ids (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) (gigl.types.graph.LoadedGraphTensors attribute) node_label_column_name (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) node_labels (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite.ToyGraphData attribute) (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) node_partition_book (gigl.types.graph.PartitionOutput attribute) node_pb (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) node_preprocessor_config (snapchat.research.gbml.gigl_resource_config_pb2.DataPreprocessorConfig property) NODE_PREPROCESSOR_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.DataPreprocessorConfig attribute) node_type (gigl.src.data_preprocessor.lib.ingest.reference.NodeDataReference attribute) node_type_lin_dict (gigl.src.common.models.pyg.heterogeneous.SimpleHGN attribute) node_type_to_condensed_node_type_map (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) node_type_to_inferencer_output_info_map (snapchat.research.gbml.inference_metadata_pb2.InferenceMetadata property) NODE_TYPE_TO_INFERENCER_OUTPUT_INFO_MAP_FIELD_NUMBER (snapchat.research.gbml.inference_metadata_pb2.InferenceMetadata attribute) node_type_to_random_negative_tfrecord_uri_prefix (snapchat.research.gbml.flattened_graph_metadata_pb2.NodeAnchorBasedLinkPredictionOutput property) NODE_TYPE_TO_RANDOM_NEGATIVE_TFRECORD_URI_PREFIX_FIELD_NUMBER (snapchat.research.gbml.flattened_graph_metadata_pb2.NodeAnchorBasedLinkPredictionOutput attribute) node_types (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) (gigl.src.common.types.pb_wrappers.graph_metadata.GraphMetadataPbWrapper property) (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite.ToyGraphData attribute) (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo property) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata property) NODE_TYPES_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.GraphMetadata attribute) NodeAnchorBasedLinkPredictionBaseInferencer (class in gigl.src.inference.v1.lib.base_inferencer) NodeAnchorBasedLinkPredictionBaseTask (class in gigl.src.common.models.layers.task) NodeAnchorBasedLinkPredictionBatch (class in gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader) NodeAnchorBasedLinkPredictionBatch.BatchSupervisionEdgeData (class in gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader) NodeAnchorBasedLinkPredictionDataset (class in snapchat.research.gbml.dataset_metadata_pb2) NodeAnchorBasedLinkPredictionDataset.TestNodeTypeToRandomNegativeDataUriEntry (class in snapchat.research.gbml.dataset_metadata_pb2) NodeAnchorBasedLinkPredictionDataset.TrainNodeTypeToRandomNegativeDataUriEntry (class in snapchat.research.gbml.dataset_metadata_pb2) NodeAnchorBasedLinkPredictionDataset.ValNodeTypeToRandomNegativeDataUriEntry (class in snapchat.research.gbml.dataset_metadata_pb2) NodeAnchorBasedLinkPredictionDatasetDataloaders (class in gigl.src.common.types.pb_wrappers.dataset_metadata_utils) NodeAnchorBasedLinkPredictionInferenceBlueprint (class in gigl.src.inference.v1.lib.node_anchor_based_link_prediction_inferencer) NodeAnchorBasedLinkPredictionModelingTaskSpec (class in gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec) NodeAnchorBasedLinkPredictionOutput (class in snapchat.research.gbml.flattened_graph_metadata_pb2) NodeAnchorBasedLinkPredictionOutput.NodeTypeToRandomNegativeTfrecordUriPrefixEntry (class in snapchat.research.gbml.flattened_graph_metadata_pb2) NodeAnchorBasedLinkPredictionSample (class in gigl.src.common.translators.training_samples_protos_translator) (class in snapchat.research.gbml.training_samples_schema_pb2) NodeAnchorBasedLinkPredictionSample.SampleSupervisionEdgeData (class in gigl.src.common.translators.training_samples_protos_translator) NodeAnchorBasedLinkPredictionTaskInputs (class in gigl.src.common.types.task_inputs) NodeAnchorBasedLinkPredictionTasks (class in gigl.src.common.models.layers.task) NodeAnchorLinkSplitter (class in gigl.utils.data_splitters) NodeClassificationInferenceBlueprint (class in gigl.src.inference.v1.lib.node_classification_inferencer) NodeClassificationModelingTaskSpec (class in gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec) NodeDataPreprocessingSpec (class in gigl.src.data_preprocessor.lib.types) NodeDataReference (class in gigl.src.data_preprocessor.lib.ingest.reference) NodeId (in module gigl.src.common.types.graph_data) NodeOutputIdentifier (class in gigl.src.data_preprocessor.lib.types) NodePbWrapper (class in gigl.src.common.types.pb_wrappers.graph_data_types) NodePbWrapperCoder (class in gigl.common.utils.compute.serialization.serialize_protos) nodes (snapchat.research.gbml.graph_schema_pb2.Graph property) NODES_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.Graph attribute) nodes_pb (gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper property) nodes_wrapper (gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper property) NodeType (in module gigl.src.common.types.graph_data) NopMetricsPublisher (class in gigl.common.metrics.base_metrics) num_edge_features (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo property) num_edge_types (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) num_edges (gigl.src.data_preprocessor.lib.enumerate.utils.EnumeratorEdgeTypeMetadata attribute) num_epochs (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec attribute) num_heads (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec attribute) (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) num_hops (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.GlobalRandomUniformStrategy attribute) NUM_HOPS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.GlobalRandomUniformStrategy attribute) num_layers (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec attribute) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec attribute) (gigl.src.common.models.pyg.heterogeneous.SimpleHGN attribute) (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) num_lines_in_file() (in module gigl.common.utils.os_utils) num_local_ssds (gigl.src.common.utils.spark_job_manager.DataprocClusterInitData attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SparkResourceConfig attribute) NUM_LOCAL_SSDS_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.SparkResourceConfig attribute) num_max_training_samples_to_output (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) NUM_MAX_TRAINING_SAMPLES_TO_OUTPUT_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) num_neighbors_to_sample (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) NUM_NEIGHBORS_TO_SAMPLE_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) num_node_distinct_labels (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo property) num_node_features (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo property) num_nodes (gigl.src.data_preprocessor.lib.enumerate.utils.EnumeratorNodeTypeMetadata attribute) (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo property) num_nodes_to_sample (gigl.src.common.types.pb_wrappers.sampling_op.RandomUniformPbWrapper property) (gigl.src.common.types.pb_wrappers.sampling_op.RandomWeightedPbWrapper property) (gigl.src.common.types.pb_wrappers.sampling_op.TopKPbWrapper property) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.RandomUniform attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.RandomWeighted attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.TopK attribute) NUM_NODES_TO_SAMPLE_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.RandomUniform attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.RandomWeighted attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.TopK attribute) num_parallel_file_reads (gigl.common.data.dataloaders.TFDatasetOptions attribute) num_partitions (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) num_positive_samples (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) NUM_POSITIVE_SAMPLES_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) num_replicas (snapchat.research.gbml.gigl_resource_config_pb2.KFPResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.KFPTrainerConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SparkResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiTrainerConfig attribute) NUM_REPLICAS_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.KFPResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.KFPTrainerConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SparkResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiTrainerConfig attribute) num_test_batches (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec attribute) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec attribute) num_unique_query_ids (gigl.src.common.types.task_inputs.BatchCombinedScores attribute) num_user_def_edge_features (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo property) num_user_defined_negative_samples (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) NUM_USER_DEFINED_NEGATIVE_SAMPLES_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) num_user_defined_positive_samples (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) NUM_USER_DEFINED_POSITIVE_SAMPLES_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) num_val_batches (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec attribute) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec attribute) num_workers (gigl.src.common.utils.spark_job_manager.DataprocClusterInitData attribute) (gigl.src.training.v1.lib.data_loaders.common.DataloaderConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.DataflowResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.LocalResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.LocalTrainerConfig attribute) NUM_WORKERS_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.DataflowResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.LocalResourceConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.LocalTrainerConfig attribute) NumpyCoder (class in gigl.common.utils.compute.serialization.serialize_np) O offline_encoder (gigl.src.common.models.layers.task.BGRL attribute) (gigl.src.common.models.layers.task.TBGRL attribute) OFFLINE_METRIC_PREFIX (in module gigl.src.common.constants.metrics) op_name (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper property) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp attribute) OP_NAME_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp attribute) op_name_to_sampling_op_pb_wrapper (gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathPbWrapper property) OpsMetricPublisher (class in gigl.common.metrics.metrics_interface) optim_lr (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec attribute) out_channels (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec attribute) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec attribute) (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) out_dim (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) (gigl.src.common.models.pyg.nn.models.feature_embedding.FeatureEmbeddingLayer property) (gigl.src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge attribute) out_lin (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) OUTGOING (gigl.src.common.types.pb_wrappers.sampling_op.SamplingDirectionEnumType attribute) (in module snapchat.research.gbml.subgraph_sampling_strategy_pb2) output_dim (gigl.src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction property) output_metadata (gigl.src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper property) (gigl.src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper property) output_metadata_type (gigl.src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper property) (gigl.src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper property) P p_rel (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) PADDING_NODE (in module gigl.utils.data_splitters) parallel (gigl.src.common.models.pyg.nn.models.feature_interaction.CombinationMode attribute) params (gigl.src.common.types.pb_wrappers.sampling_op.UserDefinedPbWrapper property) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.UserDefined property) PARAMS_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.UserDefined attribute) parent_sampling_op_pb_wrappers (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper property) parse_args() (in module gigl.common.utils.test_utils) parse_bq_table_path() (gigl.src.common.utils.bq.BqUtils static method) parse_formatted_datetime() (in module gigl.src.common.utils.time) parse_makefile_vars() (in module gigl.common.constants) parser (in module gigl.orchestration.kubeflow.runner) (in module gigl.src.config_populator.config_populator) (in module gigl.src.data_preprocessor.data_preprocessor) (in module gigl.src.inference.inferencer) (in module gigl.src.inference.v1.gnn_inferencer) (in module gigl.src.inference.v2.glt_inferencer) (in module gigl.src.mocking.dataset_asset_mocking_suite) (in module gigl.src.post_process.post_processor) (in module gigl.src.split_generator.split_generator) (in module gigl.src.subgraph_sampler.subgraph_sampler) (in module gigl.src.training.trainer) (in module gigl.src.training.v1.lib.training_process) (in module gigl.src.training.v1.trainer) (in module gigl.src.training.v2.glt_trainer) (in module gigl.src.validation_check.config_validator) partition() (gigl.distributed.dist_partitioner.DistPartitioner method) (gigl.distributed.DistPartitioner method) partition_edge_index_and_edge_features() (gigl.distributed.dist_partitioner.DistPartitioner method) (gigl.distributed.dist_range_partitioner.DistRangePartitioner method) (gigl.distributed.DistPartitioner method) (gigl.distributed.DistRangePartitioner method) partition_idx (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) partition_labels() (gigl.distributed.dist_partitioner.DistPartitioner method) (gigl.distributed.DistPartitioner method) partition_node() (gigl.distributed.dist_partitioner.DistPartitioner method) (gigl.distributed.DistPartitioner method) partition_node_features() (gigl.distributed.dist_partitioner.DistPartitioner method) (gigl.distributed.DistPartitioner method) partitioned_edge_features (gigl.types.graph.PartitionOutput attribute) partitioned_edge_index (gigl.types.graph.PartitionOutput attribute) partitioned_negative_labels (gigl.types.graph.PartitionOutput attribute) partitioned_node_features (gigl.types.graph.PartitionOutput attribute) partitioned_positive_labels (gigl.types.graph.PartitionOutput attribute) PartitionOutput (class in gigl.types.graph) PassthroughCoder (class in gigl.common.beam.coders) PassthroughPreprocessorConfigForMockedAssets (class in gigl.src.mocking.mocking_assets.passthrough_preprocessor_config_for_mocked_assets) path (gigl.common.GcsUri property) (gigl.common.types.uri.gcs_uri.GcsUri property) PATH_BASE_IMAGES_VARIABLE_FILE (in module gigl.common.constants) PATH_GIGL_PKG_INIT_FILE (in module gigl.common.constants) path_to_udf (gigl.src.common.types.pb_wrappers.sampling_op.UserDefinedPbWrapper property) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.UserDefined attribute) PATH_TO_UDF_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.UserDefined attribute) paths (snapchat.research.gbml.subgraph_sampling_strategy_pb2.MessagePassingPathStrategy property) PATHS_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.MessagePassingPathStrategy attribute) pb (gigl.src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper attribute) (gigl.src.common.types.pb_wrappers.graph_data_types.GraphPbWrapper attribute) (gigl.src.common.types.pb_wrappers.graph_data_types.NodePbWrapper attribute) perm_assets_bucket (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) PERM_ASSETS_BUCKET_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) perm_assets_bucket_path (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) pin_memory (gigl.src.training.v1.lib.data_loaders.common.DataloaderConfig attribute) PIPELINE_COMPONENT_PREFIX (in module gigl.src.common.constants.metrics) PipelineConfig (class in gigl.orchestration.local.runner) pod_name (gigl.common.types.wrappers.kfp_api.KfpTaskDetails property) pos_edge_feats (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite.UserDefinedLabels attribute) pos_edge_features (gigl.src.common.translators.training_samples_protos_translator.NodeAnchorBasedLinkPredictionSample.SampleSupervisionEdgeData attribute) pos_edge_index (gigl.src.mocking.dataset_asset_mocking_suite.DatasetAssetMockingSuite.UserDefinedLabels attribute) pos_edges (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample property) POS_EDGES_FIELD_NUMBER (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample attribute) pos_embeddings (gigl.src.common.types.task_inputs.BatchEmbeddings attribute) pos_nodes (gigl.src.common.translators.training_samples_protos_translator.NodeAnchorBasedLinkPredictionSample.SampleSupervisionEdgeData attribute) pos_scores (gigl.src.common.types.task_inputs.BatchScores attribute) pos_supervision_edge_data (gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader.NodeAnchorBasedLinkPredictionBatch attribute) POSITIVE (gigl.src.common.types.graph_data.EdgeUsageType attribute) positive_edge_info (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput property) POSITIVE_EDGE_INFO_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput attribute) positive_edge_label (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) positive_ids (gigl.src.common.types.task_inputs.BatchCombinedScores attribute) positive_label (gigl.types.graph.LoadedGraphTensors attribute) positive_label_entity_info (gigl.common.data.load_torch_tensors.SerializedGraphMetadata attribute) post_processor_args (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.PostProcessorConfig property) POST_PROCESSOR_ARGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.PostProcessorConfig attribute) post_processor_cls_path (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.PostProcessorConfig attribute) POST_PROCESSOR_CLS_PATH_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.PostProcessorConfig attribute) post_processor_config (snapchat.research.gbml.gbml_config_pb2.GbmlConfig property) POST_PROCESSOR_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig attribute) post_processor_log_metrics_uri (snapchat.research.gbml.postprocessed_metadata_pb2.PostProcessedMetadata attribute) POST_PROCESSOR_LOG_METRICS_URI_FIELD_NUMBER (snapchat.research.gbml.postprocessed_metadata_pb2.PostProcessedMetadata attribute) postprocessed_metadata (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig property) POSTPROCESSED_METADATA_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) PostProcessedMetadata (class in snapchat.research.gbml.postprocessed_metadata_pb2) PostProcessor (class in gigl.src.post_process.post_processor) (gigl.src.common.constants.components.GiGLComponents attribute) PREDICTION_TAGGED_OUTPUT_KEY (in module gigl.src.inference.v1.lib.base_inference_blueprint) predictions (gigl.src.inference.v1.lib.base_inferencer.InferBatchResults attribute) predictions_path (snapchat.research.gbml.inference_metadata_pb2.InferenceOutput attribute) PREDICTIONS_PATH_FIELD_NUMBER (snapchat.research.gbml.inference_metadata_pb2.InferenceOutput attribute) prepare_for_pipeline() (gigl.src.data_preprocessor.lib.data_preprocessor_config.DataPreprocessorConfig method) prepare_staging_paths() (gigl.src.inference.lib.assets.InferenceAssets static method) preprocess_node_anchor_based_link_prediction_raw_sample_fn() (gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader.NodeAnchorBasedLinkPredictionBatch static method) preprocess_node_anchor_based_link_prediction_sample_fn() (gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader.NodeAnchorBasedLinkPredictionBatch static method) preprocess_node_classification_raw_sample_fn() (gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader.SupervisedNodeClassificationBatch static method) preprocess_node_classification_sample_fn() (gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader.SupervisedNodeClassificationBatch static method) preprocess_rooted_node_neighborhood_raw_sample_fn() (gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader.RootedNodeNeighborhoodBatch static method) preprocess_rooted_node_neighborhood_sample_fn() (gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader.RootedNodeNeighborhoodBatch static method) preprocessed_metadata (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper property) preprocessed_metadata_pb (gigl.src.common.types.pb_wrappers.preprocessed_metadata.PreprocessedMetadataPbWrapper attribute) preprocessed_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) preprocessed_metadata_uri (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) PREPROCESSED_METADATA_URI_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) PreprocessedMetadata (class in snapchat.research.gbml.preprocessed_metadata_pb2) PreprocessedMetadata.CondensedEdgeTypeToPreprocessedMetadataEntry (class in snapchat.research.gbml.preprocessed_metadata_pb2) PreprocessedMetadata.CondensedNodeTypeToPreprocessedMetadataEntry (class in snapchat.research.gbml.preprocessed_metadata_pb2) PreprocessedMetadata.EdgeMetadataInfo (class in snapchat.research.gbml.preprocessed_metadata_pb2) PreprocessedMetadata.EdgeMetadataOutput (class in snapchat.research.gbml.preprocessed_metadata_pb2) PreprocessedMetadata.NodeMetadataOutput (class in snapchat.research.gbml.preprocessed_metadata_pb2) PreprocessedMetadataPbWrapper (class in gigl.src.common.types.pb_wrappers.preprocessed_metadata) PreprocessedMetadataReferences (class in gigl.src.data_preprocessor.data_preprocessor) preprocessing_fn (gigl.src.data_preprocessor.lib.transform.utils.AnalyzeAndBuildTFTransformFn attribute) (gigl.src.data_preprocessor.lib.types.EdgeDataPreprocessingSpec attribute) (gigl.src.data_preprocessor.lib.types.NodeDataPreprocessingSpec attribute) preprocessor_config (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig property) PREPROCESSOR_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig attribute) pretrained_tft_model_uri (gigl.src.data_preprocessor.lib.types.EdgeDataPreprocessingSpec attribute) (gigl.src.data_preprocessor.lib.types.NodeDataPreprocessingSpec attribute) process() (gigl.common.beam.coders.RecordBatchToTFExampleCoderFn method) (gigl.common.beam.coders.RuntimeTFExampleProtoCoderFn method) (gigl.common.logger.Logger method) (gigl.src.data_preprocessor.lib.transform.utils.InstanceDictToTFExample method) (gigl.src.inference.v1.lib.transforms.batch_generator.BatchProcessorDoFn method) process_raw_pyg_samples_and_collate_fn() (gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader.RootedNodeNeighborhoodBatch static method) (gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader.SupervisedNodeClassificationBatch static method) profile_memory (gigl.src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler attribute) profileit() (in module gigl.common.metrics.decorators) profiler_args (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig property) PROFILER_ARGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig attribute) profiler_config (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig property) PROFILER_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig attribute) profiler_context() (gigl.src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler method) profiler_log_dir (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig attribute) PROFILER_LOG_DIR_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig attribute) project (gigl.common.services.vertex_ai.VertexAIService property) (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) (gigl.src.common.utils.spark_job_manager.DataprocClusterInitData attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) PROJECT_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) project_id (gigl.common.services.dataproc.DataprocService attribute) proto_coder (gigl.common.utils.compute.serialization.serialize_protos.EdgePbWrapperCoder attribute) (gigl.common.utils.compute.serialization.serialize_protos.GraphPbWrapperCoder attribute) (gigl.common.utils.compute.serialization.serialize_protos.NodePbWrapperCoder attribute) ProtoUtils (class in gigl.common.utils.proto_utils) PYG (gigl.src.common.types.model.GraphBackend attribute) PygGraphBuilder (class in gigl.src.common.graph_builder.pyg_graph_builder) PygGraphData (class in gigl.src.common.graph_builder.pyg_graph_data) Q query_embeddings (gigl.src.common.types.task_inputs.BatchEmbeddings attribute) R ram_budget_multiplier (gigl.common.data.dataloaders.TFDatasetOptions attribute) random_neg_batch (gigl.src.common.types.task_inputs.InputBatch attribute) random_neg_embeddings (gigl.src.common.types.task_inputs.BatchEmbeddings attribute) random_neg_ids (gigl.src.common.types.task_inputs.BatchCombinedScores attribute) random_neg_scores (gigl.src.common.types.task_inputs.BatchScores attribute) random_negative_batch_size (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec attribute) random_uniform (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp property) RANDOM_UNIFORM_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp attribute) random_uniform_pb (gigl.src.common.types.pb_wrappers.sampling_op.RandomUniformPbWrapper attribute) random_uniform_spec (snapchat.research.gbml.subgraph_sampling_strategy_pb2.GlobalRandomUniformStrategy property) RANDOM_UNIFORM_SPEC_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.GlobalRandomUniformStrategy attribute) random_weighted (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp property) RANDOM_WEIGHTED_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp attribute) random_weighted_pb (gigl.src.common.types.pb_wrappers.sampling_op.RandomWeightedPbWrapper attribute) RandomUniform (class in snapchat.research.gbml.subgraph_sampling_strategy_pb2) RandomUniformPbWrapper (class in gigl.src.common.types.pb_wrappers.sampling_op) RandomWeighted (class in snapchat.research.gbml.subgraph_sampling_strategy_pb2) RandomWeightedPbWrapper (class in gigl.src.common.types.pb_wrappers.sampling_op) raw_data_schema_file_path (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) RawBatchType (in module gigl.src.inference.v1.lib.transforms.batch_generator) RawSampleType (in module gigl.src.inference.v1.lib.base_inference_blueprint) read_args() (gigl.src.common.modeling_task_specs.utils.parameter_loading.ArgumentReader method) read_from_gcs() (gigl.common.utils.gcs.GcsUtils method) read_proto_from_binary() (gigl.common.utils.proto_utils.ProtoUtils method) read_proto_from_yaml() (gigl.common.utils.proto_utils.ProtoUtils method) read_training_sample_protos_from_tfrecords() (in module gigl.src.common.types.pb_wrappers.dataset_metadata_utils) ReadExistingTFTransformFn (class in gigl.src.data_preprocessor.lib.transform.utils) reconstruction_decoder (gigl.src.common.models.layers.task.FeatureReconstruction attribute) reconstruction_enc_dec (gigl.src.common.models.layers.task.FeatureReconstruction attribute) reconstruction_mask (gigl.src.common.models.layers.task.FeatureReconstruction attribute) record_shapes (gigl.src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler attribute) RecordBatchToTFExampleCoderFn (class in gigl.common.beam.coders) reference_uri (gigl.src.data_preprocessor.lib.ingest.reference.DataReference attribute) region (gigl.common.services.dataproc.DataprocService attribute) (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) (gigl.src.common.utils.spark_job_manager.DataprocClusterInitData attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) REGION_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) register_edge_features() (gigl.distributed.dist_partitioner.DistPartitioner method) (gigl.distributed.DistPartitioner method) register_edge_index() (gigl.distributed.dist_partitioner.DistPartitioner method) (gigl.distributed.dist_range_partitioner.DistRangePartitioner method) (gigl.distributed.DistPartitioner method) (gigl.distributed.DistRangePartitioner method) register_edge_types() (gigl.src.common.graph_builder.abstract_graph_builder.GraphBuilder method) register_labels() (gigl.distributed.dist_partitioner.DistPartitioner method) (gigl.distributed.DistPartitioner method) register_node_features() (gigl.distributed.dist_partitioner.DistPartitioner method) (gigl.distributed.DistPartitioner method) register_node_field() (gigl.src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchemaBuilder method) register_node_ids() (gigl.distributed.dist_partitioner.DistPartitioner method) (gigl.distributed.DistPartitioner method) relation (gigl.src.common.types.graph_data.EdgeType attribute) Relation (in module gigl.src.common.types.graph_data) relation (snapchat.research.gbml.graph_schema_pb2.EdgeType attribute) RELATION_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.EdgeType attribute) remove_file_if_exist() (in module gigl.common.utils.local_fs) remove_file_or_folder_if_exist() (in module gigl.common.utils.local_fs) remove_folder_if_exist() (in module gigl.common.utils.local_fs) remove_line_from_file() (in module gigl.common.utils.local_fs) repeat (gigl.src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler attribute) repeated_candidate_scores (gigl.src.common.types.task_inputs.BatchCombinedScores attribute) REPEATED_OP_NAME (gigl.src.common.types.exception.SubgraphSamplingValidationErrorType attribute) repeated_query_embeddings (gigl.src.common.types.task_inputs.BatchEmbeddings attribute) repeated_query_ids (gigl.src.common.types.task_inputs.BatchCombinedScores attribute) REPEATED_ROOT_NODE_TYPE (gigl.src.common.types.exception.SubgraphSamplingValidationErrorType attribute) replica_count (gigl.common.services.vertex_ai.VertexAiJobConfig attribute) reset() (gigl.src.common.graph_builder.abstract_graph_builder.GraphBuilder method) (gigl.src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchemaBuilder method) reset_parameters() (gigl.src.common.models.layers.feature_interaction.DCNCross method) (gigl.src.common.models.layers.feature_interaction.DCNv2 method) (gigl.src.common.models.pyg.nn.conv.edge_attr_gat_conv.EdgeAttrGATConv method) (gigl.src.common.models.pyg.nn.conv.gin_conv.GINEConv method) (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv method) (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv method) (gigl.src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction method) (gigl.src.common.models.pyg.nn.models.jumping_knowledge.JumpingKnowledge method) resource_config (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper attribute) (in module gigl.env.pipelines_config) RESOURCE_CONFIG_OS_ENV (in module gigl.src.common.constants.resource_config) resource_config_uri (gigl.orchestration.local.runner.PipelineConfig attribute) resource_labels (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig property) RESOURCE_LABELS_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig attribute) result_types (gigl.src.common.models.layers.task.BGRL property) (gigl.src.common.models.layers.task.DirectAU property) (gigl.src.common.models.layers.task.FeatureReconstruction property) (gigl.src.common.models.layers.task.GBT property) (gigl.src.common.models.layers.task.GRACE property) (gigl.src.common.models.layers.task.Margin property) (gigl.src.common.models.layers.task.NodeAnchorBasedLinkPredictionBaseTask property) (gigl.src.common.models.layers.task.NodeAnchorBasedLinkPredictionTasks property) (gigl.src.common.models.layers.task.Retrieval property) (gigl.src.common.models.layers.task.Softmax property) (gigl.src.common.models.layers.task.TBGRL property) (gigl.src.common.models.layers.task.WhiteningDecorrelation property) RetriesFailedException Retrieval (class in gigl.src.common.models.layers.task) RetrievalLoss (class in gigl.src.common.models.layers.loss) retry() (in module gigl.common.utils.retry) return_emb (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) root_edge (snapchat.research.gbml.training_samples_schema_pb2.SupervisedLinkBasedTaskSample property) ROOT_EDGE_FIELD_NUMBER (snapchat.research.gbml.training_samples_schema_pb2.SupervisedLinkBasedTaskSample attribute) root_edge_labels (snapchat.research.gbml.training_samples_schema_pb2.SupervisedLinkBasedTaskSample property) ROOT_EDGE_LABELS_FIELD_NUMBER (snapchat.research.gbml.training_samples_schema_pb2.SupervisedLinkBasedTaskSample attribute) root_node (gigl.src.common.translators.training_samples_protos_translator.NodeAnchorBasedLinkPredictionSample attribute) (gigl.src.common.translators.training_samples_protos_translator.RootedNodeNeighborhoodSample attribute) (gigl.src.common.translators.training_samples_protos_translator.SupervisedNodeClassificationSample attribute) (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample property) (snapchat.research.gbml.training_samples_schema_pb2.RootedNodeNeighborhood property) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedNodeClassificationSample property) ROOT_NODE_FIELD_NUMBER (snapchat.research.gbml.training_samples_schema_pb2.NodeAnchorBasedLinkPredictionSample attribute) (snapchat.research.gbml.training_samples_schema_pb2.RootedNodeNeighborhood attribute) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedNodeClassificationSample attribute) root_node_indices (gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader.NodeAnchorBasedLinkPredictionBatch attribute) (gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader.SupervisedNodeClassificationBatch attribute) root_node_labels (gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader.SupervisedNodeClassificationBatch attribute) (snapchat.research.gbml.training_samples_schema_pb2.SupervisedNodeClassificationSample property) ROOT_NODE_LABELS_FIELD_NUMBER (snapchat.research.gbml.training_samples_schema_pb2.SupervisedNodeClassificationSample attribute) root_node_to_target_node_id (gigl.src.training.v1.lib.data_loaders.node_anchor_based_link_prediction_data_loader.NodeAnchorBasedLinkPredictionBatch.BatchSupervisionEdgeData attribute) root_node_type (gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathPbWrapper property) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.MessagePassingPath attribute) ROOT_NODE_TYPE_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.MessagePassingPath attribute) ROOT_NODE_TYPE_NOT_IN_GRAPH_METADATA (gigl.src.common.types.exception.SubgraphSamplingValidationErrorType attribute) ROOT_NODE_TYPE_NOT_IN_TASK_METADATA (gigl.src.common.types.exception.SubgraphSamplingValidationErrorType attribute) root_node_type_to_message_passing_path_pb_wrapper (gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy.GlobalRandomUniformStrategyPbWrapper property) (gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathStrategyPbWrapper property) (gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy.SubgraphSamplingStrategyPbWrapper property) root_nodes (gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader.RootedNodeNeighborhoodBatch attribute) (gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader.SupervisedNodeClassificationBatch attribute) root_sampling_op_names (gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy.MessagePassingPathPbWrapper property) RootedNodeNeighborhood (class in snapchat.research.gbml.training_samples_schema_pb2) RootedNodeNeighborhoodBatch (class in gigl.src.training.v1.lib.data_loaders.rooted_node_neighborhood_data_loader) RootedNodeNeighborhoodSample (class in gigl.src.common.translators.training_samples_protos_translator) RT (in module gigl.common.utils.func_tools) RUN (gigl.orchestration.kubeflow.runner.Action attribute) run() (gigl.orchestration.kubeflow.kfp_orchestrator.KfpOrchestrator method), [1] (gigl.orchestration.local.runner.Runner static method) (gigl.src.config_populator.config_populator.ConfigPopulator method) (gigl.src.data_preprocessor.data_preprocessor.DataPreprocessor method) (gigl.src.data_preprocessor.lib.enumerate.utils.Enumerator method) (gigl.src.inference.inferencer.Inferencer method) (gigl.src.inference.v1.gnn_inferencer.InferencerV1 method) (gigl.src.inference.v2.glt_inferencer.GLTInferencer method) (gigl.src.post_process.post_processor.PostProcessor method) (gigl.src.split_generator.split_generator.SplitGenerator method) (gigl.src.subgraph_sampler.subgraph_sampler.SubgraphSampler method) (gigl.src.training.trainer.Trainer method) (gigl.src.training.v1.lib.training_process.GnnTrainingProcess method) (gigl.src.training.v1.trainer.Trainer method) (gigl.src.training.v2.glt_trainer.GLTTrainer method) run_command_and_stream_stdout() (in module gigl.common.utils.os_utils) run_config_populator() (gigl.orchestration.local.runner.Runner static method) run_data_preprocessor() (gigl.orchestration.local.runner.Runner static method) run_inferencer() (gigl.orchestration.local.runner.Runner static method) RUN_NO_COMPILE (gigl.orchestration.kubeflow.runner.Action attribute) run_pipeline() (gigl.common.services.vertex_ai.VertexAIService method) run_post_process() (gigl.src.post_process.lib.base_post_processor.BasePostProcessor method) run_query() (gigl.src.common.utils.bq.BqUtils method) run_split_generator() (gigl.orchestration.local.runner.Runner static method) run_subgraph_sampler() (gigl.orchestration.local.runner.Runner static method) run_tests() (in module gigl.common.utils.test_utils) run_trainer() (gigl.orchestration.local.runner.Runner static method) Runner (class in gigl.orchestration.local.runner) RuntimeTFExampleProtoCoderFn (class in gigl.common.beam.coders) S sample_edge_type (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) sample_hydrate_user_def_edge() (in module gigl.src.mocking.lib.user_defined_edge_sampling) sample_node_type (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) sampling_direction (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper property) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp attribute) SAMPLING_DIRECTION_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp attribute) sampling_method (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper property) SAMPLING_OP_EDGE_TYPE_NOT_IN_GRAPH_METADATA (gigl.src.common.types.exception.SubgraphSamplingValidationErrorType attribute) sampling_op_pb (gigl.src.common.types.pb_wrappers.sampling_op.SamplingOpPbWrapper attribute) sampling_ops (snapchat.research.gbml.subgraph_sampling_strategy_pb2.MessagePassingPath property) SAMPLING_OPS_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.MessagePassingPath attribute) SamplingDirection (class in snapchat.research.gbml.subgraph_sampling_strategy_pb2) SamplingDirectionEnumType (class in gigl.src.common.types.pb_wrappers.sampling_op) SamplingOp (class in snapchat.research.gbml.subgraph_sampling_strategy_pb2) SamplingOpPbWrapper (class in gigl.src.common.types.pb_wrappers.sampling_op) save_model() (in module gigl.src.training.v1.lib.training_process) save_scripted_model() (in module gigl.src.common.utils.model) save_state_dict() (in module gigl.src.common.utils.model) schema (gigl.src.data_preprocessor.lib.transform.utils.IngestRawFeatures attribute) (gigl.src.data_preprocessor.lib.transform.utils.InstanceDictToTFExample attribute) (gigl.src.data_preprocessor.lib.transform.utils.WriteTFSchema attribute) (gigl.src.data_preprocessor.lib.types.FeatureSchema attribute) (gigl.src.inference.v1.lib.inference_output_schema.InferenceOutputBigqueryTableSchema attribute) schema_descriptor (gigl.src.data_preprocessor.lib.transform.utils.WriteTFSchema attribute) schema_uri (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) SCHEMA_URI_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) score() (gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec.NodeClassificationModelingTaskSpec method) scripted_model_uri (snapchat.research.gbml.trained_model_metadata_pb2.TrainedModelMetadata attribute) SCRIPTED_MODEL_URI_FIELD_NUMBER (snapchat.research.gbml.trained_model_metadata_pb2.TrainedModelMetadata attribute) SECONDS (gigl.common.metrics.decorators.TimerRecordGranularity attribute) seed (gigl.src.training.v1.lib.data_loaders.common.DataloaderConfig attribute) select_label_edge_types() (in module gigl.types.graph) select_ssl_positive_label_edges() (in module gigl.utils.data_splitters) serialize_edge_features() (gigl.src.common.utils.data.feature_serialization.FeatureSerializationUtils class method) serialize_node_features() (gigl.src.common.utils.data.feature_serialization.FeatureSerializationUtils class method) SerializedGraphMetadata (class in gigl.common.data.load_torch_tensors) SerializedTFRecordInfo (class in gigl.common.data.dataloaders) service_account (gigl.src.common.utils.spark_job_manager.DataprocClusterInitData attribute) service_account_email (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) setup_for_training() (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec method) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec method) (gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec.NodeClassificationModelingTaskSpec method) (gigl.src.training.v1.lib.base_trainer.BaseTrainer method) setup_model_device() (in module gigl.src.training.v1.lib.training_process) shape (gigl.common.utils.compute.serialization.serialize_np.EncodedNdArray attribute) share_ipc() (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset method) (gigl.distributed.DistLinkPredictionDataset method) share_memory() (in module gigl.utils.share_memory) shared_args (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig property) SHARED_ARGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) shared_config (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig property) SHARED_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig attribute) shared_resource_config (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig property) SHARED_RESOURCE_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig attribute) shared_resource_config_uri (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig attribute) SHARED_RESOURCE_CONFIG_URI_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig attribute) SharedResourceConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) SharedResourceConfig.CommonComputeConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) SharedResourceConfig.ResourceLabelsEntry (class in snapchat.research.gbml.gigl_resource_config_pb2) should_distribute() (in module gigl.common.utils.torch_training) should_enable_candidate_sampling_correction (gigl.src.common.models.layers.task.Retrieval attribute) should_enable_profiler (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig attribute) SHOULD_ENABLE_PROFILER_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig attribute) should_have_edge_features (gigl.src.common.models.pyg.heterogeneous.SimpleHGN attribute) should_include_isolated_nodes_in_training (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) SHOULD_INCLUDE_ISOLATED_NODES_IN_TRAINING_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) should_l2_normalize_embedding_layer_output (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec attribute) (gigl.src.common.models.pyg.heterogeneous.HGT attribute) (gigl.src.common.models.pyg.heterogeneous.SimpleHGN attribute) (gigl.src.common.models.pyg.homogeneous.BasicHomogeneousGNN attribute) should_log_to_tensorboard (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig attribute) SHOULD_LOG_TO_TENSORBOARD_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig attribute) should_loop (gigl.src.training.v1.lib.data_loaders.common.DataloaderConfig attribute) should_normalize (gigl.src.common.models.pyg.homogeneous.TwoLayerGCN attribute) should_populate_predictions_path (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) should_skip_automatic_temp_asset_cleanup (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) SHOULD_SKIP_AUTOMATIC_TEMP_ASSET_CLEANUP_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) should_skip_inference (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) SHOULD_SKIP_INFERENCE_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) should_skip_model_evaluation (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) SHOULD_SKIP_MODEL_EVALUATION_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) should_skip_training (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) SHOULD_SKIP_TRAINING_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) should_use_glt_backend (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) SimpleHGN (class in gigl.src.common.models.pyg.heterogeneous) SimpleHGNConv (class in gigl.src.common.models.pyg.nn.conv.simplehgn_conv) skip (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) snapchat module snapchat.research module snapchat.research.gbml module snapchat.research.gbml.dataset_metadata_pb2 module snapchat.research.gbml.flattened_graph_metadata_pb2 module snapchat.research.gbml.gbml_config_pb2 module snapchat.research.gbml.gigl_resource_config_pb2 module snapchat.research.gbml.graph_schema_pb2 module snapchat.research.gbml.inference_metadata_pb2 module snapchat.research.gbml.postprocessed_metadata_pb2 module snapchat.research.gbml.preprocessed_metadata_pb2 module snapchat.research.gbml.subgraph_sampling_strategy_pb2 module snapchat.research.gbml.trained_model_metadata_pb2 module snapchat.research.gbml.training_samples_schema_pb2 module Softmax (class in gigl.src.common.models.layers.task) softmax_temperature (gigl.src.common.models.layers.loss.SoftmaxLoss attribute) SoftmaxLoss (class in gigl.src.common.models.layers.loss) SortedDict (class in gigl.common.collections.sorted_dict) SPARK_31_TFRECORD_JAR_GCS_PATH (in module gigl.common.constants) SPARK_31_TFRECORD_JAR_LOCAL_PATH (in module gigl.common.constants) SPARK_35_TFRECORD_JAR_GCS_PATH (in module gigl.common.constants) SPARK_35_TFRECORD_JAR_LOCAL_PATH (in module gigl.common.constants) SPARK_SPLIT_GENERATOR_CONFIG (in module gigl.src.common.constants.resource_config) SparkJobManager (class in gigl.src.common.utils.spark_job_manager) SparkResourceConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) SPECED_COMPONENTS (in module gigl.orchestration.kubeflow.kfp_pipeline) split_and_write_node_anchor_link_prediction_subgraph_samples_from_mocked_dataset_info() (in module gigl.src.mocking.lib.mock_input_for_trainer) split_and_write_supervised_node_classification_subgraph_samples_from_mocked_dataset_info() (in module gigl.src.mocking.lib.mock_input_for_trainer) split_generator_config (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig property) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig property) SPLIT_GENERATOR_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig attribute) split_strategy_args (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig property) SPLIT_STRATEGY_ARGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig attribute) split_strategy_cls_path (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig attribute) SPLIT_STRATEGY_CLS_PATH_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig attribute) SplitGenerator (class in gigl.src.split_generator.split_generator) (gigl.src.common.constants.components.GiGLComponents attribute) src_identifier (gigl.src.data_preprocessor.lib.ingest.reference.EdgeDataReference attribute) src_node (gigl.src.common.types.graph_data.Edge property) (gigl.src.data_preprocessor.lib.types.EdgeOutputIdentifier attribute) src_node_id (gigl.src.common.types.graph_data.Edge attribute) (gigl.src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper property) (snapchat.research.gbml.graph_schema_pb2.Edge attribute) SRC_NODE_ID_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.Edge attribute) src_node_id_key (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput attribute) SRC_NODE_ID_KEY_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput attribute) src_node_type (gigl.src.common.types.graph_data.EdgeType attribute) (snapchat.research.gbml.graph_schema_pb2.EdgeType attribute) SRC_NODE_TYPE_FIELD_NUMBER (snapchat.research.gbml.graph_schema_pb2.EdgeType attribute) stacked (gigl.src.common.models.pyg.nn.models.feature_interaction.CombinationMode attribute) START_COMPONENT_TO_ASSET_CHECKS_MAP (in module gigl.src.validation_check.config_validator) START_COMPONENT_TO_CLS_CHECKS_MAP (in module gigl.src.validation_check.config_validator) START_COMPONENT_TO_RESOURCE_CONFIG_CHECKS_MAP (in module gigl.src.validation_check.config_validator) START_STOP_COMPONENT_TO_CLS_CHECKS_MAP (in module gigl.src.validation_check.config_validator) START_STOP_COMPONENT_TO_RESOURCE_CONFIG_CHECKS_MAP (in module gigl.src.validation_check.config_validator) started_at (gigl.common.types.wrappers.kfp_api.KfpTaskDetails property) stats_file_path (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) stats_output_uri (gigl.src.data_preprocessor.lib.transform.utils.GenerateAndVisualizeStats attribute) step() (gigl.src.common.modeling_task_specs.utils.early_stop.EarlyStopper method) StringCoder (class in gigl.common.utils.compute.serialization.serialize_string) subgraph (gigl.src.common.translators.training_samples_protos_translator.NodeAnchorBasedLinkPredictionSample attribute) (gigl.src.common.translators.training_samples_protos_translator.RootedNodeNeighborhoodSample attribute) subgraph_node_to_global_node_mapping (gigl.src.common.graph_builder.gbml_graph_protocol.GbmlGraphDataProtocol property) (gigl.src.common.graph_builder.pyg_graph_data.PygGraphData property) subgraph_sampler_config (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig property) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig property) SUBGRAPH_SAMPLER_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig attribute) subgraph_sampling_strategy (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig property) SUBGRAPH_SAMPLING_STRATEGY_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig attribute) subgraph_sampling_strategy_pb (gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy.SubgraphSamplingStrategyPbWrapper attribute) subgraph_sampling_strategy_pb_wrapper (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) SubgraphSampler (class in gigl.src.subgraph_sampler.subgraph_sampler) (gigl.src.common.constants.components.GiGLComponents attribute) SubgraphSamplingStrategy (class in snapchat.research.gbml.subgraph_sampling_strategy_pb2) SubgraphSamplingStrategyPbWrapper (class in gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy) SubgraphSamplingValidationError SubgraphSamplingValidationErrorType (class in gigl.src.common.types.exception) submit_and_wait_scala_spark_job() (gigl.common.services.dataproc.DataprocService method) (gigl.src.common.utils.spark_job_manager.SparkJobManager method) supervised_link_based_task_dataset (snapchat.research.gbml.dataset_metadata_pb2.DatasetMetadata property) SUPERVISED_LINK_BASED_TASK_DATASET_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.DatasetMetadata attribute) supervised_link_based_task_output (snapchat.research.gbml.flattened_graph_metadata_pb2.FlattenedGraphMetadata property) SUPERVISED_LINK_BASED_TASK_OUTPUT_FIELD_NUMBER (snapchat.research.gbml.flattened_graph_metadata_pb2.FlattenedGraphMetadata attribute) supervised_node_classification_dataset (snapchat.research.gbml.dataset_metadata_pb2.DatasetMetadata property) SUPERVISED_NODE_CLASSIFICATION_DATASET_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.DatasetMetadata attribute) supervised_node_classification_output (snapchat.research.gbml.flattened_graph_metadata_pb2.FlattenedGraphMetadata property) SUPERVISED_NODE_CLASSIFICATION_OUTPUT_FIELD_NUMBER (snapchat.research.gbml.flattened_graph_metadata_pb2.FlattenedGraphMetadata attribute) SupervisedLinkBasedTaskOutput (class in snapchat.research.gbml.flattened_graph_metadata_pb2) SupervisedLinkBasedTaskSample (class in snapchat.research.gbml.training_samples_schema_pb2) SupervisedLinkBasedTaskSplitDataset (class in snapchat.research.gbml.dataset_metadata_pb2) SupervisedNodeClassificationBaseInferencer (class in gigl.src.inference.v1.lib.base_inferencer) SupervisedNodeClassificationBatch (class in gigl.src.training.v1.lib.data_loaders.supervised_node_classification_data_loader) SupervisedNodeClassificationDataset (class in snapchat.research.gbml.dataset_metadata_pb2) SupervisedNodeClassificationDatasetDataloaders (class in gigl.src.common.types.pb_wrappers.dataset_metadata_utils) SupervisedNodeClassificationOutput (class in snapchat.research.gbml.flattened_graph_metadata_pb2) SupervisedNodeClassificationSample (class in gigl.src.common.translators.training_samples_protos_translator) (class in snapchat.research.gbml.training_samples_schema_pb2) supervision_edge_types (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata property) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata property) SUPERVISION_EDGE_TYPES_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata.LinkBasedTaskMetadata attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata.NodeAnchorBasedLinkPredictionTaskMetadata attribute) supervision_node_types (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata property) SUPERVISION_NODE_TYPES_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata.NodeBasedTaskMetadata attribute) supports_distributed_training (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec property) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec property) (gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec.NodeClassificationModelingTaskSpec property) (gigl.src.training.v1.lib.base_trainer.BaseTrainer property) supports_edge_attr (gigl.src.common.models.pyg.homogeneous.EdgeAttrGAT attribute) (gigl.src.common.models.pyg.homogeneous.GAT attribute) (gigl.src.common.models.pyg.homogeneous.GATv2 attribute) (gigl.src.common.models.pyg.homogeneous.GIN attribute) (gigl.src.common.models.pyg.homogeneous.GINE attribute) (gigl.src.common.models.pyg.homogeneous.GraphSAGE attribute) (gigl.src.common.models.pyg.homogeneous.Transformer attribute) supports_edge_weight (gigl.src.common.models.pyg.homogeneous.EdgeAttrGAT attribute) (gigl.src.common.models.pyg.homogeneous.GAT attribute) (gigl.src.common.models.pyg.homogeneous.GATv2 attribute) (gigl.src.common.models.pyg.homogeneous.GIN attribute) (gigl.src.common.models.pyg.homogeneous.GINE attribute) (gigl.src.common.models.pyg.homogeneous.GraphSAGE attribute) (gigl.src.common.models.pyg.homogeneous.Transformer attribute) T T (in module gigl.common.collections.itertools) (in module gigl.common.utils.compute.serialization.coder) (in module gigl.common.utils.proto_utils) (in module gigl.common.utils.retry) (in module gigl.src.common.types.dataset_split) (in module gigl.src.common.types.pb_wrappers.dataset_metadata_utils) (in module gigl.src.inference.v1.lib.base_inferencer) (in module gigl.src.mocking.lib.tfrecord_io) (in module gigl.src.training.v1.lib.data_loaders.tf_records_iterable_dataset) target_uri (gigl.src.data_preprocessor.lib.transform.utils.WriteTFSchema attribute) task_config_uri (gigl.orchestration.local.runner.PipelineConfig attribute) task_details_map (gigl.common.types.wrappers.kfp_api.ApiRunDetailWrapper property) task_metadata (gigl.src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper property) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig property) TASK_METADATA_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig attribute) task_metadata_pb (gigl.src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper attribute) task_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) (gigl.src.config_populator.config_populator.ConfigPopulator property) task_metadata_type (gigl.src.common.types.pb_wrappers.task_metadata.TaskMetadataPbWrapper property) (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) task_name (gigl.src.common.models.layers.task.NodeAnchorBasedLinkPredictionBaseTask property) TaskMetadataPb (in module gigl.src.common.types.pb_wrappers.types) TaskMetadataPbWrapper (class in gigl.src.common.types.pb_wrappers.task_metadata) TaskMetadataType (class in gigl.src.common.types.task_metadata) tasks (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec attribute) (gigl.src.common.models.pyg.link_prediction.LinkPredictionGNN property) TBGRL (class in gigl.src.common.models.layers.task) TBGRLLoss (class in gigl.src.common.models.layers.loss) temp_assets_bq_dataset_name (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) TEMP_ASSETS_BQ_DATASET_NAME_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) temp_assets_bucket (gigl.src.common.utils.spark_job_manager.DataprocClusterInitData attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) TEMP_ASSETS_BUCKET_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) temp_assets_bucket_path (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) temp_assets_regional_bucket_path (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) temp_embedding_gcs_path (gigl.src.inference.v1.gnn_inferencer.InferencerOutputPaths attribute) temp_predictions_gcs_path (gigl.src.inference.v1.gnn_inferencer.InferencerOutputPaths attribute) temp_regional_assets_bucket (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) TEMP_REGIONAL_ASSETS_BUCKET_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.CommonComputeConfig attribute) temperature (gigl.src.common.models.layers.loss.GRACELoss attribute) (gigl.src.common.models.layers.loss.UniformityLoss attribute) template_gbml_config (gigl.src.config_populator.config_populator.ConfigPopulator property) tensor_adapter_config (gigl.src.data_preprocessor.lib.transform.utils.AnalyzeAndBuildTFTransformFn attribute) tensorboard_logs_uri (snapchat.research.gbml.trained_model_metadata_pb2.TrainedModelMetadata attribute) TENSORBOARD_LOGS_URI_FIELD_NUMBER (snapchat.research.gbml.trained_model_metadata_pb2.TrainedModelMetadata attribute) TEST (gigl.src.common.types.dataset_split.DatasetSplit attribute) test_data_uri (snapchat.research.gbml.dataset_metadata_pb2.SupervisedLinkBasedTaskSplitDataset attribute) (snapchat.research.gbml.dataset_metadata_pb2.SupervisedNodeClassificationDataset attribute) TEST_DATA_URI_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.SupervisedLinkBasedTaskSplitDataset attribute) (snapchat.research.gbml.dataset_metadata_pb2.SupervisedNodeClassificationDataset attribute) test_file_pattern (gigl.common.utils.test_utils.TestArgs attribute), [1] test_main (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.Dataloaders attribute) (gigl.src.training.v1.lib.data_loaders.common.DataloaderTypes attribute) test_main_data_uri (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset attribute) TEST_MAIN_DATA_URI_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset attribute) test_node_ids (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) test_node_type_to_random_negative_data_uri (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset property) TEST_NODE_TYPE_TO_RANDOM_NEGATIVE_DATA_URI_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset attribute) test_random_negative (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.Dataloaders attribute) (gigl.src.training.v1.lib.data_loaders.common.DataloaderTypes attribute) TestArgs (class in gigl.common.utils.test_utils) tf_on_cpu() (in module gigl.common.utils.decorator) tf_transform_directory (gigl.src.data_preprocessor.lib.transform.utils.ReadExistingTFTransformFn attribute) TFDatasetOptions (class in gigl.common.data.dataloaders) tfrecord_uri_pattern (gigl.common.data.dataloaders.SerializedTFRecordInfo attribute) tfrecord_uri_prefix (gigl.common.data.dataloaders.SerializedTFRecordInfo attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.NodeAnchorBasedLinkPredictionOutput attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) TFRECORD_URI_PREFIX_FIELD_NUMBER (snapchat.research.gbml.flattened_graph_metadata_pb2.NodeAnchorBasedLinkPredictionOutput attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) TFRecordDataLoader (class in gigl.common.data.dataloaders) TfRecordsIterableDataset (class in gigl.src.training.v1.lib.data_loaders.tf_records_iterable_dataset) tft_temp_directory_path (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) TFTensorDict (in module gigl.src.data_preprocessor.lib.types) TFValueEncoder (class in gigl.src.data_preprocessor.lib.transform.tf_value_encoder) TGraph (in module gigl.src.common.graph_builder.abstract_graph_builder) TimedOutException timeout (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiResourceConfig attribute) timeout() (in module gigl.src.common.utils.timeout) TIMEOUT_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.VertexAiResourceConfig attribute) timeout_s (gigl.common.services.vertex_ai.VertexAiJobConfig attribute) TIMER_CONFIG_POPULATOR_S (in module gigl.src.common.constants.metrics) TIMER_INFERENCER_S (in module gigl.src.common.constants.metrics) TIMER_POST_PROCESSOR_S (in module gigl.src.common.constants.metrics) TIMER_PREPROCESSOR_S (in module gigl.src.common.constants.metrics) TIMER_S (in module gigl.src.common.constants.metrics) TIMER_SPLIT_GENERATOR_S (in module gigl.src.common.constants.metrics) TIMER_SUBGRAPH_SAMPLER_S (in module gigl.src.common.constants.metrics) TIMER_TRAINER_CLEANUP_ENV_S (in module gigl.src.common.constants.metrics) TIMER_TRAINER_EXPORT_INFERENCE_ASSETS_S (in module gigl.src.common.constants.metrics) TIMER_TRAINER_S (in module gigl.src.common.constants.metrics) TIMER_TRAINER_SETUP_ENV_S (in module gigl.src.common.constants.metrics) TimerRecordGranularity (class in gigl.common.metrics.decorators) timers (gigl.common.metrics.base_metrics.NopMetricsPublisher attribute) TMP_PROFILER_LOG_DIR_NAME (in module gigl.src.common.modeling_task_specs.utils.profiler_wrapper) to_hetero_data() (gigl.src.common.graph_builder.pyg_graph_data.PygGraphData method) to_hetero_feat() (in module gigl.src.common.models.utils.torch) to_heterogeneous_edge() (in module gigl.types.graph) to_heterogeneous_node() (in module gigl.types.graph) to_homogeneous() (in module gigl.types.graph) top_k (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp property) TOP_K_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp attribute) top_k_pb (gigl.src.common.types.pb_wrappers.sampling_op.TopKPbWrapper attribute) TopK (class in snapchat.research.gbml.subgraph_sampling_strategy_pb2) TopKPbWrapper (class in gigl.src.common.types.pb_wrappers.sampling_op) TorchProfiler (class in gigl.src.common.modeling_task_specs.utils.profiler_wrapper) total() (gigl.src.common.models.layers.count_min_sketch.CountMinSketch method) total_steps (gigl.src.common.utils.scheduler.CosineDecayScheduler attribute) TOY_GRAPH_NODE_ANCHOR_MOCKED_DATASET_INFO (in module gigl.src.mocking.mocking_assets.mocked_datasets_for_pipeline_tests) TOY_GRAPH_USER_DEFINED_NODE_ANCHOR_MOCKED_DATASET_INFO (in module gigl.src.mocking.mocking_assets.mocked_datasets_for_pipeline_tests) trace_handler (gigl.src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler attribute) tracing_schedule (gigl.src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler attribute) TRAIN (gigl.src.common.types.dataset_split.DatasetSplit attribute) train() (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec method) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec method) (gigl.src.common.modeling_task_specs.node_classification_modeling_task_spec.NodeClassificationModelingTaskSpec method) (gigl.src.training.v1.lib.base_trainer.BaseTrainer method) train_data_uri (snapchat.research.gbml.dataset_metadata_pb2.SupervisedLinkBasedTaskSplitDataset attribute) (snapchat.research.gbml.dataset_metadata_pb2.SupervisedNodeClassificationDataset attribute) TRAIN_DATA_URI_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.SupervisedLinkBasedTaskSplitDataset attribute) (snapchat.research.gbml.dataset_metadata_pb2.SupervisedNodeClassificationDataset attribute) train_main (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.Dataloaders attribute) (gigl.src.training.v1.lib.data_loaders.common.DataloaderTypes attribute) train_main_data_uri (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset attribute) TRAIN_MAIN_DATA_URI_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset attribute) train_model() (in module gigl.src.mocking.lib.mock_input_for_inference) train_node_ids (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) train_node_type_to_random_negative_data_uri (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset property) TRAIN_NODE_TYPE_TO_RANDOM_NEGATIVE_DATA_URI_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset attribute) train_random_negative (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.Dataloaders attribute) (gigl.src.training.v1.lib.data_loaders.common.DataloaderTypes attribute) trained_model_metadata (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig property) TRAINED_MODEL_METADATA_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig attribute) trained_model_metadata_pb (gigl.src.common.types.pb_wrappers.trained_model_metadata.TrainedModelMetadataPbWrapper attribute) trained_model_metadata_pb_wrapper (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) trained_model_uri (snapchat.research.gbml.trained_model_metadata_pb2.TrainedModelMetadata attribute) TRAINED_MODEL_URI_FIELD_NUMBER (snapchat.research.gbml.trained_model_metadata_pb2.TrainedModelMetadata attribute) TrainedModelMetadata (class in snapchat.research.gbml.trained_model_metadata_pb2) TrainedModelMetadataPbWrapper (class in gigl.src.common.types.pb_wrappers.trained_model_metadata) Trainer (class in gigl.src.training.trainer) (class in gigl.src.training.v1.trainer) (gigl.src.common.constants.components.GiGLComponents attribute) trainer_args (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig property) TRAINER_ARGS_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig attribute) trainer_cls_path (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig attribute) TRAINER_CLS_PATH_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig attribute) trainer_config (gigl.src.common.types.pb_wrappers.gbml_config.GbmlConfigPbWrapper property) (gigl.src.common.types.pb_wrappers.gigl_resource_config.GiglResourceConfigWrapper property) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig property) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig property) TRAINER_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gbml_config_pb2.GbmlConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig attribute) trainer_resource_config (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig property) TRAINER_RESOURCE_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig attribute) TrainerResourceConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) training_sample_type (gigl.src.common.types.pb_wrappers.dataset_metadata.DatasetMetadataPbWrapper property) (gigl.src.common.types.pb_wrappers.flattened_graph_metadata.FlattenedGraphMetadataPbWrapper property) training_samples_from_NodeAnchorBasedLinkPredictionSamplePb() (gigl.src.common.translators.training_samples_protos_translator.TrainingSamplesProtosTranslator static method) training_samples_from_RootedNodeNeighborhoodPb() (gigl.src.common.translators.training_samples_protos_translator.TrainingSamplesProtosTranslator static method) training_samples_from_SupervisedNodeClassificationSamplePb() (gigl.src.common.translators.training_samples_protos_translator.TrainingSamplesProtosTranslator static method) TrainingSamplePb (in module gigl.src.common.types.pb_wrappers.types) TrainingSamplesProtosTranslator (class in gigl.src.common.translators.training_samples_protos_translator) TrainingSamplesSchemaProtoUtils (class in gigl.src.common.utils.gbml_protos) transform_directory_path (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) transform_fn_assets_uri (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) TRANSFORM_FN_ASSETS_URI_FIELD_NUMBER (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput attribute) transformed_features_file_prefix (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) transformed_features_schema_path (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) transformed_features_transform_fn_assets_path (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) TransformedFeaturesInfo (class in gigl.src.data_preprocessor.lib.transform.transformed_features_info) Transformer (class in gigl.src.common.models.pyg.homogeneous) treat_labels_as_edges() (gigl.types.graph.LoadedGraphTensors method) tuple_repr() (gigl.src.common.types.graph_data.EdgeType method) TwoLayerGCN (class in gigl.src.common.models.pyg.homogeneous) type (gigl.src.common.types.graph_data.Node attribute) U unenumerate_all_inferred_bq_assets() (in module gigl.src.post_process.utils.unenumeration) UnenumerateAssets (class in gigl.src.inference.v1.lib.utils) UNENUMERATED_EMBEDDINGS_TABLE_SCHEMA (in module gigl.src.inference.v1.lib.inference_output_schema) UNENUMERATED_PREDICTIONS_TABLE_SCHEMA (in module gigl.src.inference.v1.lib.inference_output_schema) UNENUMERATION_QUERY (in module gigl.src.inference.v1.lib.queries) uniformity_loss (gigl.src.common.models.layers.task.DirectAU attribute) UniformityLoss (class in gigl.src.common.models.layers.loss) unique_id (gigl.src.common.types.pb_wrappers.graph_data_types.EdgePbWrapper property) (gigl.src.common.types.pb_wrappers.graph_data_types.NodePbWrapper property) UNIQUE_NODE_ENUMERATION_QUERY (in module gigl.src.data_preprocessor.lib.enumerate.queries) unlabeled_tfrecord_uri_prefix (snapchat.research.gbml.flattened_graph_metadata_pb2.SupervisedLinkBasedTaskOutput attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.SupervisedNodeClassificationOutput attribute) UNLABELED_TFRECORD_URI_PREFIX_FIELD_NUMBER (snapchat.research.gbml.flattened_graph_metadata_pb2.SupervisedLinkBasedTaskOutput attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.SupervisedNodeClassificationOutput attribute) unprocessed_datasets_gcs_uri (in module gigl.src.mocking.lib.pyg_datasets_forks) update_bq_dataset_retention() (gigl.src.common.utils.bq.BqUtils method) update_bq_table_retention() (gigl.src.common.utils.bq.BqUtils method) update_bq_table_with_test_assets_and_version() (in module gigl.src.mocking.lib.constants) update_gcs_uri_with_test_assets_and_version() (in module gigl.src.mocking.lib.constants) update_mocked_dataset_artifact_metadata() (in module gigl.src.mocking.lib.versioning) update_offline_encoder() (gigl.src.common.models.layers.task.BGRL method) (gigl.src.common.models.layers.task.TBGRL method) upload() (gigl.orchestration.kubeflow.kfp_orchestrator.KfpOrchestrator method) upload_files_to_gcs() (gigl.common.utils.gcs.GcsUtils method) upload_from_filelike() (gigl.common.utils.gcs.GcsUtils method) upload_from_string() (gigl.common.utils.gcs.GcsUtils method) UPLOAD_RETRY_DEADLINE_S (in module gigl.common.utils.gcs) Uri (class in gigl.common) (class in gigl.common.types.uri.uri) uri (gigl.common.types.uri.uri.Uri property) (gigl.common.Uri property) UriFactory (class in gigl.common) (class in gigl.common.types.uri.uri_factory) uris (gigl.src.training.v1.lib.data_loaders.common.DataloaderConfig attribute) UriType (in module gigl.src.training.v1.lib.data_loaders.tf_records_iterable_dataset) url (gigl.src.mocking.lib.pyg_datasets_forks.CoraFromGCS attribute) (gigl.src.mocking.lib.pyg_datasets_forks.DBLPFromGCS attribute) use_dcnv2_feats_interaction (gigl.src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction attribute) use_interleave (gigl.common.data.dataloaders.TFDatasetOptions attribute) use_mlp_feats_interaction (gigl.src.common.models.pyg.nn.models.feature_interaction.FeatureInteraction attribute) user_defined (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp property) user_defined_edge_feats (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) user_defined_edge_index (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) USER_DEFINED_FIELD_NUMBER (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp attribute) user_defined_pb (gigl.src.common.types.pb_wrappers.sampling_op.UserDefinedPbWrapper attribute) UserDefined (class in snapchat.research.gbml.subgraph_sampling_strategy_pb2) UserDefined.ParamsEntry (class in snapchat.research.gbml.subgraph_sampling_strategy_pb2) UserDefinedPbWrapper (class in gigl.src.common.types.pb_wrappers.sampling_op) V v_rel (gigl.src.common.models.pyg.nn.conv.hgt_conv.HGTConv attribute) VAL (gigl.src.common.types.dataset_split.DatasetSplit attribute) val_data_uri (snapchat.research.gbml.dataset_metadata_pb2.SupervisedLinkBasedTaskSplitDataset attribute) (snapchat.research.gbml.dataset_metadata_pb2.SupervisedNodeClassificationDataset attribute) VAL_DATA_URI_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.SupervisedLinkBasedTaskSplitDataset attribute) (snapchat.research.gbml.dataset_metadata_pb2.SupervisedNodeClassificationDataset attribute) val_main (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.Dataloaders attribute) (gigl.src.training.v1.lib.data_loaders.common.DataloaderTypes attribute) val_main_data_uri (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset attribute) VAL_MAIN_DATA_URI_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset attribute) val_node_ids (gigl.distributed.dist_link_prediction_dataset.DistLinkPredictionDataset property) (gigl.distributed.DistLinkPredictionDataset property) val_node_type_to_random_negative_data_uri (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset property) VAL_NODE_TYPE_TO_RANDOM_NEGATIVE_DATA_URI_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset attribute) val_random_negative (gigl.src.common.types.pb_wrappers.dataset_metadata_utils.Dataloaders attribute) (gigl.src.training.v1.lib.data_loaders.common.DataloaderTypes attribute) validate() (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec method) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec method) validate_dags() (gigl.src.common.types.pb_wrappers.subgraph_sampling_strategy.SubgraphSamplingStrategyPbWrapper method) validate_every_n_batches (gigl.src.common.modeling_task_specs.graphsage_template_modeling_spec.GraphSageTemplateTrainerSpec attribute) (gigl.src.common.modeling_task_specs.node_anchor_based_link_prediction_modeling_task_spec.NodeAnchorBasedLinkPredictionModelingTaskSpec attribute) value (gigl.src.common.types.model_eval_metrics.EvalMetric attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.TestNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.TrainNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.ValNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.NodeAnchorBasedLinkPredictionOutput.NodeTypeToRandomNegativeTfrecordUriPrefixEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.FeatureFlagsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDbArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig.InferencerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.MetricsConfig.MetricsArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig.ProfilerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig.SharedArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig.TrainerArgsEntry attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.ResourceLabelsEntry attribute) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedEdgeTypeMapEntry property) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedNodeTypeMapEntry attribute) (snapchat.research.gbml.inference_metadata_pb2.InferenceMetadata.NodeTypeToInferencerOutputInfoMapEntry property) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedEdgeTypeToPreprocessedMetadataEntry property) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedNodeTypeToPreprocessedMetadataEntry property) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.UserDefined.ParamsEntry attribute) VALUE_FIELD_NUMBER (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.TestNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.TrainNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.dataset_metadata_pb2.NodeAnchorBasedLinkPredictionDataset.ValNodeTypeToRandomNegativeDataUriEntry attribute) (snapchat.research.gbml.flattened_graph_metadata_pb2.NodeAnchorBasedLinkPredictionOutput.NodeTypeToRandomNegativeTfrecordUriPrefixEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.DataPreprocessorConfig.DataPreprocessorArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig.AssignerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SplitGeneratorConfig.SplitStrategyArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.DatasetConfig.SubgraphSamplerConfig.ExperimentalFlagsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.FeatureFlagsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDbArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.GraphDBConfig.GraphDbIngestionArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig.InferencerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.MetricsConfig.MetricsArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.PostProcessorConfig.PostProcessorArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.ProfilerConfig.ProfilerArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.SharedConfig.SharedArgsEntry attribute) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig.TrainerArgsEntry attribute) (snapchat.research.gbml.gigl_resource_config_pb2.SharedResourceConfig.ResourceLabelsEntry attribute) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedEdgeTypeMapEntry attribute) (snapchat.research.gbml.graph_schema_pb2.GraphMetadata.CondensedNodeTypeMapEntry attribute) (snapchat.research.gbml.inference_metadata_pb2.InferenceMetadata.NodeTypeToInferencerOutputInfoMapEntry attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedEdgeTypeToPreprocessedMetadataEntry attribute) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.CondensedNodeTypeToPreprocessedMetadataEntry attribute) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.UserDefined.ParamsEntry attribute) version (gigl.src.mocking.lib.mocked_dataset_resources.MockedDatasetInfo attribute) (gigl.src.mocking.lib.versioning.MockedDatasetArtifactMetadata attribute) vertex_ai_inferencer_config (snapchat.research.gbml.gigl_resource_config_pb2.InferencerResourceConfig property) VERTEX_AI_INFERENCER_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.InferencerResourceConfig attribute) vertex_ai_trainer_config (snapchat.research.gbml.gigl_resource_config_pb2.DistributedTrainerConfig property) (snapchat.research.gbml.gigl_resource_config_pb2.TrainerResourceConfig property) VERTEX_AI_TRAINER_CONFIG_FIELD_NUMBER (snapchat.research.gbml.gigl_resource_config_pb2.DistributedTrainerConfig attribute) (snapchat.research.gbml.gigl_resource_config_pb2.TrainerResourceConfig attribute) VertexAiJobConfig (class in gigl.common.services.vertex_ai) VertexAiResourceConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) VertexAIService (class in gigl.common.services.vertex_ai) VertexAiTrainerConfig (class in snapchat.research.gbml.gigl_resource_config_pb2) visible_devices (in module gigl.src.training.v1.lib.data_loaders.tf_records_iterable_dataset) visualized_facets_file_path (gigl.src.data_preprocessor.lib.transform.transformed_features_info.TransformedFeaturesInfo attribute) VT (in module gigl.common.collections.frozen_dict) (in module gigl.common.collections.sorted_dict) W W_etype (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) W_nfeat (gigl.src.common.models.pyg.nn.conv.simplehgn_conv.SimpleHGNConv attribute) wait (gigl.src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler attribute) wait_for_completion() (gigl.orchestration.kubeflow.kfp_orchestrator.KfpOrchestrator method), [1] wait_for_run_completion() (gigl.common.services.vertex_ai.VertexAIService static method) warmup (gigl.src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler attribute) warmup_steps (gigl.src.common.utils.scheduler.CosineDecayScheduler attribute) weights (gigl.types.graph.GraphPartitionData attribute) WhichOneof() (snapchat.research.gbml.dataset_metadata_pb2.DatasetMetadata method) (snapchat.research.gbml.flattened_graph_metadata_pb2.FlattenedGraphMetadata method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.InferencerConfig method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TaskMetadata method) (snapchat.research.gbml.gbml_config_pb2.GbmlConfig.TrainerConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.DistributedTrainerConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.GiglResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.InferencerResourceConfig method) (snapchat.research.gbml.gigl_resource_config_pb2.TrainerResourceConfig method) (snapchat.research.gbml.graph_schema_pb2.Edge method) (snapchat.research.gbml.graph_schema_pb2.Node method) (snapchat.research.gbml.inference_metadata_pb2.InferenceOutput method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataInfo method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.EdgeMetadataOutput method) (snapchat.research.gbml.preprocessed_metadata_pb2.PreprocessedMetadata.NodeMetadataOutput method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SamplingOp method) (snapchat.research.gbml.subgraph_sampling_strategy_pb2.SubgraphSamplingStrategy method) WhiteningDecorrelation (class in gigl.src.common.models.layers.task) WhiteningDecorrelationLoss (class in gigl.src.common.models.layers.loss) with_stack (gigl.src.common.modeling_task_specs.utils.profiler_wrapper.TorchProfiler attribute) write_from_gcs_to_bq() (gigl.src.inference.v1.gnn_inferencer.InferencerV1 method) write_kfp_metrics_to_pipeline_metric_path() (gigl.src.common.translators.model_eval_metrics_translator.EvalMetricsCollectionTranslator class method) write_pb_tfrecord_shards_to_uri() (in module gigl.src.mocking.lib.tfrecord_io) write_proto_to_binary() (gigl.common.utils.proto_utils.ProtoUtils method) write_proto_to_yaml() (gigl.common.utils.proto_utils.ProtoUtils method) write_to_output_viewer_path() (gigl.common.utils.kfp.KfpOutputViewers method) WriteTFSchema (class in gigl.src.data_preprocessor.lib.transform.utils) X x (gigl.src.common.translators.training_samples_protos_translator.SupervisedNodeClassificationSample attribute) Y y (gigl.src.common.translators.training_samples_protos_translator.SupervisedNodeClassificationSample attribute) yield_instance_dict_ptransform() (gigl.src.data_preprocessor.lib.ingest.bigquery.BigqueryEdgeDataReference method) (gigl.src.data_preprocessor.lib.ingest.bigquery.BigqueryNodeDataReference method) (gigl.src.data_preprocessor.lib.ingest.reference.DataReference method)
