Expose integrated gradient function to enable building custom exporte…

…r with it. PiperOrigin-RevId: 507884056
tensorflow · Feb 7, 2023 · 811d907 · 811d907
1 parent 44eef67
commit 811d907
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow_gnn/runner/__init__.py b/tensorflow_gnn/runner/__init__.py
@@ -53,6 +53,7 @@
 
 # Model helpers
 chain_first_output = model_utils.chain_first_output
+integrated_gradients = attribution.integrated_gradients
 
 # Orchestration
 run = orchestration.run