Optimize einsum when node[1].is_const() (#1739)

Signed-off-by: xavier dupré <[email protected]> Co-authored-by: xavier dupré <[email protected]>
onnx · Oct 8, 2021 · 4245d8d · 4245d8d
1 parent 5770433
commit 4245d8d
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tf2onnx/optimizer/einsum_optimizer.py b/tf2onnx/optimizer/einsum_optimizer.py
@@ -2291,8 +2291,6 @@ def _optimize_at_current_graph_level(self, graph):
 
     def _optimize_einsum(self, node, graph):
         """Apply the optimizer."""
-        if node.inputs[1].is_const():
-            return False
         inp_shape = graph.get_shape(node.input[0])
         if inp_shape is None:
             # The rank must be known