export nodes belonging to inner lambdas

1 files changed, 7 insertions, 1 deletions

diff --git a/eval/src/vespa/eval/eval/node_types.cpp b/eval/src/vespa/eval/eval/node_types.cpp
index cbc96e719e0..6c50280741c 100644
--- a/eval/src/vespa/eval/eval/node_types.cpp
+++ b/eval/src/vespa/eval/eval/node_types.cpp
@@ -307,7 +307,13 @@ struct TypeExporter : public NodeTraverser {
         : parent_type_map(parent_type_map_in),
           exported_type_map(exported_type_map_out),
           missing_cnt(0) {}
-    bool open(const Node &) override { return true; }
+    bool open(const Node &node) override {
+        if (auto lambda = dynamic_cast<const TensorLambda *>(&node)) {
+            lambda->lambda().root().traverse(*this);
+            return false;
+        }
+        return true;
+    }
     void close(const Node &node) override {
         auto pos = parent_type_map.find(&node);
         if (pos != parent_type_map.end()) {