AUTHORS
BUILD
LICENSE
MANIFEST.in
README.md
setup.py
./WORKSPACE
./package/move_generated_files.sh
./package/tfdep.bzl
./tensorflow_gnn/tensorflow_gnn.bzl
package/BUILD
tensorflow_gnn/BUILD
tensorflow_gnn/__init__.py
tensorflow_gnn/version.py
tensorflow_gnn.egg-info/PKG-INFO
tensorflow_gnn.egg-info/SOURCES.txt
tensorflow_gnn.egg-info/dependency_links.txt
tensorflow_gnn.egg-info/entry_points.txt
tensorflow_gnn.egg-info/namespace_packages.txt
tensorflow_gnn.egg-info/not-zip-safe
tensorflow_gnn.egg-info/requires.txt
tensorflow_gnn.egg-info/top_level.txt
tensorflow_gnn/converters/__init__.py
tensorflow_gnn/converters/triples.py
tensorflow_gnn/converters/triples_test.py
tensorflow_gnn/converters/ogb/__init__.py
tensorflow_gnn/converters/ogb/convert_ogb_dataset.py
tensorflow_gnn/converters/ogb/ogb_lib.py
tensorflow_gnn/data/BUILD
tensorflow_gnn/data/__init__.py
tensorflow_gnn/data/unigraph.py
tensorflow_gnn/data/unigraph_test.py
tensorflow_gnn/experimental/__init__.py
tensorflow_gnn/experimental/in_memory/__init__.py
tensorflow_gnn/experimental/in_memory/datasets.py
tensorflow_gnn/experimental/in_memory/int_arithmetic_sampler.py
tensorflow_gnn/experimental/in_memory/int_arithmetic_sampler_test.py
tensorflow_gnn/experimental/in_memory/models.py
tensorflow_gnn/experimental/in_memory/reader_utils.py
tensorflow_gnn/experimental/in_memory/unigraph_data.py
tensorflow_gnn/experimental/in_memory/unigraph_data_test.py
tensorflow_gnn/graph/BUILD
tensorflow_gnn/graph/__init__.py
tensorflow_gnn/graph/adjacency.py
tensorflow_gnn/graph/adjacency_test.py
tensorflow_gnn/graph/batching_utils.py
tensorflow_gnn/graph/batching_utils_test.py
tensorflow_gnn/graph/dict_utils.py
tensorflow_gnn/graph/dict_utils_test.py
tensorflow_gnn/graph/graph_constants.py
tensorflow_gnn/graph/graph_piece.py
tensorflow_gnn/graph/graph_piece_test.py
tensorflow_gnn/graph/graph_tensor.py
tensorflow_gnn/graph/graph_tensor_encode.py
tensorflow_gnn/graph/graph_tensor_encode_test.py
tensorflow_gnn/graph/graph_tensor_io.py
tensorflow_gnn/graph/graph_tensor_io_test.py
tensorflow_gnn/graph/graph_tensor_ops.py
tensorflow_gnn/graph/graph_tensor_ops_test.py
tensorflow_gnn/graph/graph_tensor_pprint.py
tensorflow_gnn/graph/graph_tensor_pprint_test.py
tensorflow_gnn/graph/graph_tensor_random.py
tensorflow_gnn/graph/graph_tensor_random_test.py
tensorflow_gnn/graph/graph_tensor_test.py
tensorflow_gnn/graph/graph_tensor_test_utils.py
tensorflow_gnn/graph/normalization_ops.py
tensorflow_gnn/graph/normalization_ops_test.py
tensorflow_gnn/graph/padding_ops.py
tensorflow_gnn/graph/padding_ops_test.py
tensorflow_gnn/graph/preprocessing_common.py
tensorflow_gnn/graph/preprocessing_common_test.py
tensorflow_gnn/graph/schema_utils.py
tensorflow_gnn/graph/schema_utils_test.py
tensorflow_gnn/graph/schema_validation.py
tensorflow_gnn/graph/schema_validation_test.py
tensorflow_gnn/graph/tag_utils.py
tensorflow_gnn/graph/tag_utils_test.py
tensorflow_gnn/graph/tensor_utils.py
tensorflow_gnn/graph/tensor_utils_test.py
tensorflow_gnn/keras/BUILD
tensorflow_gnn/keras/__init__.py
tensorflow_gnn/keras/builders.py
tensorflow_gnn/keras/builders_test.py
tensorflow_gnn/keras/keras_e2e_test.py
tensorflow_gnn/keras/keras_tensors.py
tensorflow_gnn/keras/keras_tensors_test.py
tensorflow_gnn/keras/layers/BUILD
tensorflow_gnn/keras/layers/__init__.py
tensorflow_gnn/keras/layers/convolution_base.py
tensorflow_gnn/keras/layers/convolution_base_test.py
tensorflow_gnn/keras/layers/convolutions.py
tensorflow_gnn/keras/layers/convolutions_test.py
tensorflow_gnn/keras/layers/graph_ops.py
tensorflow_gnn/keras/layers/graph_ops_test.py
tensorflow_gnn/keras/layers/graph_update.py
tensorflow_gnn/keras/layers/graph_update_test.py
tensorflow_gnn/keras/layers/map_features.py
tensorflow_gnn/keras/layers/map_features_test.py
tensorflow_gnn/keras/layers/next_state.py
tensorflow_gnn/keras/layers/next_state_test.py
tensorflow_gnn/keras/layers/padding_ops.py
tensorflow_gnn/keras/layers/padding_ops_test.py
tensorflow_gnn/keras/layers/parse_example.py
tensorflow_gnn/keras/layers/parse_example_test.py
tensorflow_gnn/models/gat_v2/BUILD
tensorflow_gnn/models/gat_v2/__init__.py
tensorflow_gnn/models/gat_v2/layers.py
tensorflow_gnn/models/gat_v2/layers_test.py
tensorflow_gnn/models/gcn/BUILD
tensorflow_gnn/models/gcn/__init__.py
tensorflow_gnn/models/gcn/gcn_conv.py
tensorflow_gnn/models/gcn/gcn_conv_test.py
tensorflow_gnn/models/graph_sage/BUILD
tensorflow_gnn/models/graph_sage/__init__.py
tensorflow_gnn/models/graph_sage/layers.py
tensorflow_gnn/models/graph_sage/layers_test.py
tensorflow_gnn/models/multi_head_attention/BUILD
tensorflow_gnn/models/multi_head_attention/__init__.py
tensorflow_gnn/models/multi_head_attention/layers.py
tensorflow_gnn/models/multi_head_attention/layers_test.py
tensorflow_gnn/models/vanilla_mpnn/BUILD
tensorflow_gnn/models/vanilla_mpnn/__init__.py
tensorflow_gnn/models/vanilla_mpnn/config_dict.py
tensorflow_gnn/models/vanilla_mpnn/config_dict_test.py
tensorflow_gnn/models/vanilla_mpnn/hparams_vizier.py
tensorflow_gnn/models/vanilla_mpnn/hparams_vizier_test.py
tensorflow_gnn/models/vanilla_mpnn/layers.py
tensorflow_gnn/models/vanilla_mpnn/layers_test.py
tensorflow_gnn/proto/BUILD
tensorflow_gnn/proto/__init__.py
tensorflow_gnn/proto/examples.proto
tensorflow_gnn/proto/graph_schema.proto
tensorflow_gnn/proto/graph_schema.py
tensorflow_gnn/runner/BUILD
tensorflow_gnn/runner/__init__.py
tensorflow_gnn/runner/orchestration.py
tensorflow_gnn/runner/orchestration_test.py
tensorflow_gnn/runner/examples/ogbn/mag/BUILD
tensorflow_gnn/runner/examples/ogbn/mag/train.py
tensorflow_gnn/runner/input/BUILD
tensorflow_gnn/runner/input/datasets.py
tensorflow_gnn/runner/input/datasets_test.py
tensorflow_gnn/runner/tasks/BUILD
tensorflow_gnn/runner/tasks/classification.py
tensorflow_gnn/runner/tasks/classification_test.py
tensorflow_gnn/runner/tasks/dgi.py
tensorflow_gnn/runner/tasks/dgi_test.py
tensorflow_gnn/runner/tasks/regression.py
tensorflow_gnn/runner/tasks/regression_test.py
tensorflow_gnn/runner/trainers/BUILD
tensorflow_gnn/runner/trainers/keras_fit.py
tensorflow_gnn/runner/trainers/keras_fit_test.py
tensorflow_gnn/runner/utils/BUILD
tensorflow_gnn/runner/utils/attribution.py
tensorflow_gnn/runner/utils/attribution_test.py
tensorflow_gnn/runner/utils/model.py
tensorflow_gnn/runner/utils/model_dir.py
tensorflow_gnn/runner/utils/model_export.py
tensorflow_gnn/runner/utils/model_export_test.py
tensorflow_gnn/runner/utils/model_templates.py
tensorflow_gnn/runner/utils/model_test.py
tensorflow_gnn/runner/utils/padding.py
tensorflow_gnn/runner/utils/padding_test.py
tensorflow_gnn/runner/utils/strategies.py
tensorflow_gnn/sampler/BUILD
tensorflow_gnn/sampler/__init__.py
tensorflow_gnn/sampler/graph_sampler.py
tensorflow_gnn/sampler/graph_sampler_test.py
tensorflow_gnn/sampler/sampling_lib.py
tensorflow_gnn/sampler/sampling_lib_test.py
tensorflow_gnn/sampler/sampling_spec.proto
tensorflow_gnn/sampler/sampling_spec_builder.py
tensorflow_gnn/sampler/sampling_spec_builder_test.py
tensorflow_gnn/sampler/sampling_utils.py
tensorflow_gnn/sampler/sampling_utils_test.py
tensorflow_gnn/sampler/subgraph.proto
tensorflow_gnn/sampler/subgraph.py
tensorflow_gnn/sampler/subgraph_test.py
tensorflow_gnn/tools/BUILD
tensorflow_gnn/tools/__init__.py
tensorflow_gnn/tools/generate_training_data.py
tensorflow_gnn/tools/generate_training_data_test.py
tensorflow_gnn/tools/print_training_data.py
tensorflow_gnn/tools/print_training_data_test.py
tensorflow_gnn/tools/sampled_stats.proto
tensorflow_gnn/tools/sampled_stats.py
tensorflow_gnn/tools/sampled_stats_test.py
tensorflow_gnn/tools/validate_graph_schema.py
tensorflow_gnn/utils/BUILD
tensorflow_gnn/utils/__init__.py
tensorflow_gnn/utils/test_utils.py
testdata/BUILD
testdata/heterogeneous/BUILD
testdata/homogeneous/BUILD
testdata/node_vs_edge/BUILD