Skip to content

T
tic
Commit eed28f58 by OuHangKresnik Committed by Lianmin Zheng
Options

Fix UnboundLocalError: local variable 'tensor' referenced before assignment (#3074)

parent 3f835bd9
Showing with 2 additions and 2 deletions
topi/python/topi/opengl/pooling.py
...@@ -107,8 +107,8 @@ def schedule_pool(outs, layout): ...@@ -107,8 +107,8 @@ def schedule_pool(outs, layout):
if tag.is_broadcast(OP.tag): if tag.is_broadcast(OP.tag):
if OP not in s.outputs: if OP not in s.outputs:
s[OP].compute_inline() s[OP].compute_inline()
for tensor in OP.input_tensors and tensor.op not in scheduled_ops: for tensor in OP.input_tensors:
if tensor.op.input_tensors: if tensor.op not in scheduled_ops and tensor.op.input_tensors:
traverse(tensor.op) traverse(tensor.op)
# schedule pool # schedule pool
elif OP.tag.startswith('pool'): elif OP.tag.startswith('pool'):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment