diff options
Diffstat (limited to 'compiler/luci/pass/src/CircleTypeInferencePass.cpp')
-rw-r--r-- | compiler/luci/pass/src/CircleTypeInferencePass.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/compiler/luci/pass/src/CircleTypeInferencePass.cpp b/compiler/luci/pass/src/CircleTypeInferencePass.cpp index 67bd253e0..fb3755ffa 100644 --- a/compiler/luci/pass/src/CircleTypeInferencePass.cpp +++ b/compiler/luci/pass/src/CircleTypeInferencePass.cpp @@ -14,6 +14,8 @@ * limitations under the License. */ +#include "helpers/InferenceCandidates.h" + #include "luci/Pass/CircleTypeInferencePass.h" #include <luci/Service/CircleTypeInference.h> @@ -41,7 +43,7 @@ bool CircleTypeInferencePass::run(loco::Graph *g) luci::tinf::Rule type_infer_rule; bool changed = false; - for (auto node : loco::postorder_traversal(loco::output_nodes(g))) + for (auto node : inference_candidates(g)) { loco::DataType dtype; auto circle_node = loco::must_cast<luci::CircleNode *>(node); |