Skip to content

Commit fba20cf

Browse files
committed
refine logging, remove uesless codes
1 parent be51386 commit fba20cf

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

tf2onnx/rewriter/loop_rewriter_base.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,6 @@ def rewrite(self, context):
194194
return REWRITER_RESULT.FAIL
195195

196196
def run_internal(self):
197-
log.debug("enter loop rewriter")
198197
for op in self.g.get_nodes():
199198
if not is_loopcond_op(op):
200199
continue
@@ -278,11 +277,9 @@ def _crop_loop_body_sub_graph(self, context):
278277
output_ids = [out_tensor_value_info.id for out_tensor_value_info in outputs]
279278
ops, enter_nodes, _ = self.find_subgraph(set(input_ids), set(output_ids), self.g, merge_as_end=False)
280279

281-
other_enter_input_ids = []
282280
for enter_node in enter_nodes:
283281
# connect Enter's output to Enter's input
284282
self.g.replace_all_inputs(ops, enter_node.output[0], enter_node.input[0])
285-
other_enter_input_ids.append(enter_node.input[0])
286283

287284
return GraphInfo(ops, inputs, outputs)
288285

@@ -292,11 +289,9 @@ def _crop_loop_condition_sub_graph(self, context):
292289
outputs = [TensorValueInfo(o, self.g) for o in output_ids]
293290
ops, enter_nodes, merge_nodes = self.find_subgraph(set(input_ids), set(output_ids), self.g, merge_as_end=True)
294291

295-
other_enter_input_ids = []
296292
for enter_node in enter_nodes:
297293
# connect Enter's output to Enter's input
298294
self.g.replace_all_inputs(ops, enter_node.output[0], enter_node.input[0])
299-
other_enter_input_ids.append(enter_node.input[0])
300295

301296
dependent_vars = []
302297
for merge_node in merge_nodes:

0 commit comments

Comments
 (0)