diff --git a/src/Builder/FrontendDialectTransformer.cpp b/src/Builder/FrontendDialectTransformer.cpp index 00ab212..d7f852c 100644 --- a/src/Builder/FrontendDialectTransformer.cpp +++ b/src/Builder/FrontendDialectTransformer.cpp @@ -222,8 +222,12 @@ private: // Trailing optional inputs. if (!variadicIn) - for (auto i = inputs.size(); i < expectedNumOperands; i++) + for (auto i = inputs.size(); i < expectedNumOperands; i++) { + if (!none_) + none_ = builder_.create( + UnknownLoc(), builder_.getUnitAttr()); inputs.emplace_back(none_); + } std::vector outputTypes; @@ -482,11 +486,6 @@ private: } } - // Create a NoneTyped constant to be used for optional operation inputs - // which are not used. - none_ = - builder_.create(UnknownLoc(), builder_.getUnitAttr()); - // Import nodes in the graph. for (const auto &item : graph.node()) { ImportNode(item);