Index

_

__init__() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method) (tf_gnns.GraphTuple method) (tf_gnns.models.gcn.GCNv2 method) (tf_gnns.tfgnns_datastructures.GraphTuple method)

A

assign_global() (tf_gnns.GraphTuple method) (tf_gnns.tfgnns_datastructures.GraphTuple method)

B

build() (tf_gnns.GNCellMLP method) (tf_gnns.GraphIndep method) (tf_gnns.GraphNetMLP method) (tf_gnns.GraphNetMPNN_MLP method) (tf_gnns.models.gcn.SparseGCNConv method) (tf_gnns.models.graphnet.GNCellMLP method) (tf_gnns.models.graphnet.GraphIndep method) (tf_gnns.models.graphnet.GraphNetMLP method) (tf_gnns.models.graphnet.GraphNetMPNN_MLP method) (tf_gnns.SparseGCNConv method)

C

call() (tf_gnns.GNCellMLP method) (tf_gnns.GraphIndep method) (tf_gnns.graphnet_utils.SimpleAggLayerDense method) (tf_gnns.graphnet_utils.SimpleAggLayerSparse method) (tf_gnns.GraphNetMLP method) (tf_gnns.GraphNetMPNN_MLP method) (tf_gnns.models.gcn.GCNv2 method) (tf_gnns.models.gcn.SparseGCN method) (tf_gnns.models.gcn.SparseGCNConv method) (tf_gnns.models.graphnet.GNCellMLP method) (tf_gnns.models.graphnet.GraphIndep method) (tf_gnns.models.graphnet.GraphNetMLP method) (tf_gnns.models.graphnet.GraphNetMPNN_MLP method) (tf_gnns.SparseGCN method) (tf_gnns.SparseGCNConv method)

compare_connectivity() (tf_gnns.Graph method) (tf_gnns.tfgnns_datastructures.Graph method) copy() (tf_gnns.Edge method) (tf_gnns.Graph method) (tf_gnns.GraphTuple method) (tf_gnns.Node method) (tf_gnns.tfgnns_datastructures.Edge method) (tf_gnns.tfgnns_datastructures.Graph method) (tf_gnns.tfgnns_datastructures.GraphTuple method) (tf_gnns.tfgnns_datastructures.Node method)

E

Edge (class in tf_gnns) (class in tf_gnns.tfgnns_datastructures) EDGE_AGG_STATE (tf_gnns.graphnet_utils.GlobalInput attribute) (tf_gnns.graphnet_utils.NodeInput attribute) edge_block() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method)

EDGE_STATE (tf_gnns.graphnet_utils.EdgeInput attribute) EdgeInput (class in tf_gnns.graphnet_utils) eval_tensor_dict() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method)

G

GCNv2 (class in tf_gnns.models.gcn) get_graph() (tf_gnns.GraphTuple method) (tf_gnns.tfgnns_datastructures.GraphTuple method) get_graphnet_input_shapes() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method) get_graphnet_output_shapes() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method) get_state() (tf_gnns.Node method) (tf_gnns.tfgnns_datastructures.Node method) get_subgraph_from_nodes() (tf_gnns.Graph method) (tf_gnns.tfgnns_datastructures.Graph method) global_block() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method) GLOBAL_STATE (tf_gnns.graphnet_utils.EdgeInput attribute) (tf_gnns.graphnet_utils.GlobalInput attribute) (tf_gnns.graphnet_utils.NodeInput attribute) GlobalInput (class in tf_gnns.graphnet_utils)

GNCellMLP (class in tf_gnns) (class in tf_gnns.models.graphnet) Graph (class in tf_gnns) (class in tf_gnns.tfgnns_datastructures) graph_eval() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method) graph_tuple_eval() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method) GraphIndep (class in tf_gnns) (class in tf_gnns.models.graphnet) GraphNet (class in tf_gnns) (class in tf_gnns.graphnet_utils) GraphNetMLP (class in tf_gnns) (class in tf_gnns.models.graphnet) GraphNetMPNN_MLP (class in tf_gnns) (class in tf_gnns.models.graphnet) GraphTuple (class in tf_gnns) (class in tf_gnns.tfgnns_datastructures)

I

is_equal_by_value() (tf_gnns.Graph method) (tf_gnns.GraphTuple method) (tf_gnns.tfgnns_datastructures.Graph method) (tf_gnns.tfgnns_datastructures.GraphTuple method)

L

load() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method)

load_graph_functions() (tf_gnns.GraphNet static method) (tf_gnns.graphnet_utils.GraphNet static method)

M

make_edge_mlp() (in module tf_gnns) (in module tf_gnns.graphnet_utils) make_from_path() (tf_gnns.GraphNet static method) (tf_gnns.graphnet_utils.GraphNet static method) make_full_graphnet_functions() (in module tf_gnns) (in module tf_gnns.graphnet_utils) make_global_mlp() (in module tf_gnns) (in module tf_gnns.graphnet_utils) make_graph_indep_graphnet_functions() (in module tf_gnns) (in module tf_gnns.graphnet_utils) make_graph_to_graph_and_global_functions() (in module tf_gnns) (in module tf_gnns.graphnet_utils) make_graph_tuple_from_graph_list() (in module tf_gnns) (in module tf_gnns.tfgnns_datastructures) make_keras_simple_agg() (in module tf_gnns) (in module tf_gnns.graphnet_utils)

make_mean_max_agg() (in module tf_gnns.graphnet_utils) make_mean_max_min_agg() (in module tf_gnns.graphnet_utils) make_mean_max_min_sum_agg() (in module tf_gnns.graphnet_utils) make_mlp() (in module tf_gnns.graphnet_utils) make_mlp_graphnet_functions() (in module tf_gnns) (in module tf_gnns.graphnet_utils) make_mpnn_graphnet_noglobal_functions() (in module tf_gnns.graphnet_utils) make_node_mlp() (in module tf_gnns) (in module tf_gnns.graphnet_utils) module tf_gnns tf_gnns.graphnet_utils tf_gnns.models.gcn tf_gnns.models.graphnet tf_gnns.tfgnns_datastructures

N

Node (class in tf_gnns) (class in tf_gnns.tfgnns_datastructures) NODE_AGG_STATE (tf_gnns.graphnet_utils.GlobalInput attribute)

node_block() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method) NODE_STATE (tf_gnns.graphnet_utils.NodeInput attribute) NodeInput (class in tf_gnns.graphnet_utils)

R

RECEIVER_NODE_STATE (tf_gnns.graphnet_utils.EdgeInput attribute)

S

save() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method) scan_edge_function() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method) scan_edge_to_node_aggregation_function() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method) scan_global_function() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method) scan_node_function() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method) SENDER_NODE_STATE (tf_gnns.graphnet_utils.EdgeInput attribute)

set_tensor() (tf_gnns.Edge method) (tf_gnns.Node method) (tf_gnns.tfgnns_datastructures.Edge method) (tf_gnns.tfgnns_datastructures.Node method) SimpleAggLayerDense (class in tf_gnns.graphnet_utils) SimpleAggLayerSparse (class in tf_gnns.graphnet_utils) SparseGCN (class in tf_gnns) (class in tf_gnns.models.gcn) SparseGCNConv (class in tf_gnns) (class in tf_gnns.models.gcn) summary() (tf_gnns.GraphNet method) (tf_gnns.graphnet_utils.GraphNet method)

T

tf_gnns module tf_gnns.graphnet_utils module tf_gnns.models.gcn module

tf_gnns.models.graphnet module tf_gnns.tfgnns_datastructures module to_tensor_dict() (tf_gnns.GraphTuple method) (tf_gnns.tfgnns_datastructures.GraphTuple method)

U

unsorted_segment_max_or_zero() (in module tf_gnns.graphnet_utils) unsorted_segment_min_or_zero() (in module tf_gnns.graphnet_utils)

update_reps_for_globals() (tf_gnns.GraphTuple method) (tf_gnns.tfgnns_datastructures.GraphTuple method)

V

validate_graph() (tf_gnns.Graph static method) (tf_gnns.tfgnns_datastructures.Graph static method)