├── .gitignore ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── config.py ├── dataflow_parser.py ├── dataset_utils.py ├── experiment.py ├── layers.py ├── magnet.py ├── models.py ├── ogb_utils.py ├── presets.py ├── requirements.txt ├── run_dist_dataset_gen.sh ├── run_ogb_dataset_gen.sh ├── run_sn_dataset_gen.sh ├── script_generate_distance_np.py ├── script_generate_ogb_code2_np.py ├── script_generate_sorting_network_np.py ├── script_postcompute_eigedecomp.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/.gitignore -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/config.py -------------------------------------------------------------------------------- /dataflow_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/dataflow_parser.py -------------------------------------------------------------------------------- /dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/dataset_utils.py -------------------------------------------------------------------------------- /experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/experiment.py -------------------------------------------------------------------------------- /layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/layers.py -------------------------------------------------------------------------------- /magnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/magnet.py -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/models.py -------------------------------------------------------------------------------- /ogb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/ogb_utils.py -------------------------------------------------------------------------------- /presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/presets.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_dist_dataset_gen.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/run_dist_dataset_gen.sh -------------------------------------------------------------------------------- /run_ogb_dataset_gen.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/run_ogb_dataset_gen.sh -------------------------------------------------------------------------------- /run_sn_dataset_gen.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/run_sn_dataset_gen.sh -------------------------------------------------------------------------------- /script_generate_distance_np.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/script_generate_distance_np.py -------------------------------------------------------------------------------- /script_generate_ogb_code2_np.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/script_generate_ogb_code2_np.py -------------------------------------------------------------------------------- /script_generate_sorting_network_np.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/script_generate_sorting_network_np.py -------------------------------------------------------------------------------- /script_postcompute_eigedecomp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/script_postcompute_eigedecomp.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/digraph_transformer/HEAD/utils.py --------------------------------------------------------------------------------