path	# lines of code
MANIFEST.in	5
setup.py	136
package/tfdep.bzl	12
tensorflow_gnn/data/unigraph.py	199
tensorflow_gnn/data/__init__.py	1
tensorflow_gnn/graph/graph_tensor_test_utils.py	30
tensorflow_gnn/graph/dict_utils.py	10
tensorflow_gnn/graph/padding_ops.py	329
tensorflow_gnn/graph/tensor_utils.py	197
tensorflow_gnn/graph/graph_tensor.py	508
tensorflow_gnn/graph/schema_utils.py	98
tensorflow_gnn/graph/normalization_ops.py	62
tensorflow_gnn/graph/graph_piece.py	505
tensorflow_gnn/graph/preprocessing_common.py	11
tensorflow_gnn/graph/adjacency.py	239
tensorflow_gnn/graph/__init__.py	1
tensorflow_gnn/graph/keras/keras_tensors.py	79
tensorflow_gnn/graph/keras/builders.py	39
tensorflow_gnn/graph/keras/layers/next_state.py	79
tensorflow_gnn/graph/keras/layers/parse_example.py	23
tensorflow_gnn/graph/keras/layers/gat_v2.py	262
tensorflow_gnn/graph/keras/layers/__init__.py	33
tensorflow_gnn/graph/keras/layers/graph_ops.py	345
tensorflow_gnn/graph/keras/layers/map_features.py	142
tensorflow_gnn/graph/keras/layers/graph_update.py	303
tensorflow_gnn/graph/keras/layers/convolutions.py	69
tensorflow_gnn/graph/keras/__init__.py	6
tensorflow_gnn/graph/graph_tensor_ops.py	271
tensorflow_gnn/graph/graph_tensor_random.py	175
tensorflow_gnn/graph/schema_validation.py	208
tensorflow_gnn/graph/graph_tensor_encode.py	77
tensorflow_gnn/graph/graph_tensor_pprint.py	29
tensorflow_gnn/graph/graph_constants.py	35
tensorflow_gnn/graph/batching_utils.py	116
tensorflow_gnn/graph/graph_tensor_io.py	265
tensorflow_gnn/sampler/subgraph.proto	32
tensorflow_gnn/sampler/__init__.py	1
tensorflow_gnn/sampler/subgraph.py	103
tensorflow_gnn/tools/sampled_stats.proto	53
tensorflow_gnn/tools/sampled_stats.py	147
tensorflow_gnn/tools/generate_training_data.py	50
tensorflow_gnn/tools/__init__.py	1
tensorflow_gnn/tools/print_training_data.py	62
tensorflow_gnn/tools/validate_graph_schema.py	26
tensorflow_gnn/__init__.py	86
tensorflow_gnn/utils/__init__.py	1
tensorflow_gnn/proto/graph_schema.proto	188
tensorflow_gnn/proto/examples.proto	8
tensorflow_gnn/proto/graph_schema.py	5
tensorflow_gnn/proto/__init__.py	1
tensorflow_gnn/version.py	4