Skip to content
Snippets Groups Projects
Unverified Commit 73087901 authored by Yaman Umuroglu's avatar Yaman Umuroglu Committed by GitHub
Browse files

[Transform] call InferDataTypes in MoveScalarLinearPastInvariants

parent 4ede818c
No related branches found
No related tags found
No related merge requests found
...@@ -560,7 +560,9 @@ class MoveScalarLinearPastInvariants(Transformation): ...@@ -560,7 +560,9 @@ class MoveScalarLinearPastInvariants(Transformation):
graph_modified = True graph_modified = True
else: else:
continue continue
model = model.transform(InferShapes()) if graph_modified:
model = model.transform(InferShapes())
model = model.transform(InferDataTypes())
return (model, graph_modified) return (model, graph_modified)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment