Expose integrated gradient function to enable building custom exporte…

…r with it. PiperOrigin-RevId: 507871940
tensorflow · Feb 7, 2023 · 98e3172 · 98e3172
1 parent 44eef67
commit 98e3172
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