Skip to content
Snippets Groups Projects
Commit 2fdaf1dc authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Transform] support Div in MoveScalarLinearPastInvariants

parent 00c423dd
No related branches found
No related tags found
No related merge requests found
...@@ -533,7 +533,7 @@ class MoveScalarLinearPastInvariants(Transformation): ...@@ -533,7 +533,7 @@ class MoveScalarLinearPastInvariants(Transformation):
if prod0 is None: if prod0 is None:
continue continue
if prod0.op_type == "Mul" or prod0.op_type == "Add": if prod0.op_type in ["Mul", "Add", "Div"]:
# check if second input of producer is an initializer # check if second input of producer is an initializer
init0 = model.get_initializer(prod0.input[1]) init0 = model.get_initializer(prod0.input[1])
# if either initializer is None, skip # if either initializer is None, skip
......
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