No public description

PiperOrigin-RevId: 654665309
tensorflow · Jul 22, 2024 · 469f6d0 · 469f6d0
1 parent 95b2ac9
commit 469f6d0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorflow_gnn/experimental/sampler/proto/eval_dag.proto b/tensorflow_gnn/experimental/sampler/proto/eval_dag.proto
@@ -124,6 +124,8 @@ message Layer {
   // Names of the layer inputs. For composite layers those names must match
   // input layer names in the `eval_dag`.
   IOFeatures input_names = 5;
+  // Names of the layer outputs.
+  IOFeatures output_names = 6;
 
   // A mapping from input argument names to their spec.
   repeated ValueSpec inputs = 9;