1 files changed, 5 insertions, 1 deletions
diff --git a/nn/runtime/ExecutionBuilder.cpp b/nn/runtime/ExecutionBuilder.cpp
index 61e320f9a..d65d96446 100644
--- a/nn/runtime/ExecutionBuilder.cpp
+++ b/nn/runtime/ExecutionBuilder.cpp
@@ -876,7 +876,11 @@ std::vector<OutputShape> ExecutionBuilder::getInitialOutputShapes() const {
     std::vector<OutputShape> outputShapes(mOutputs.size());
     std::transform(mOutputs.begin(), mOutputs.end(), outputShapes.begin(),
                    [](const auto& x) -> OutputShape {
-                       return {.dimensions = x.dimensions(), .isSufficient = true};
+                       hidl_vec<uint32_t> dimensions;
+                       if (x.state() != ModelArgumentInfo::HAS_NO_VALUE) {
+                           dimensions = x.dimensions();
+                       }
+                       return {.dimensions = std::move(dimensions), .isSufficient = true};
                    });
     return outputShapes;
 }