提交 6eb0592a authored 作者: khaotik's avatar khaotik

revert change to _compile_and_check

also removed inline test case for infer_shape since the Op is removed during opt
上级 d3511f6e
......@@ -244,9 +244,9 @@ class InferShapeTester(unittest.TestCase):
# Check that the Op is removed from the compiled function.
if check_topo:
topo_shape = shapes_function.maker.fgraph.toposort()
topo_out = outputs_function.maker.fgraph.toposort()
assert not any(isinstance(t.op, cls) for t in topo_shape)
assert any(isinstance(t.op, cls) for t in topo_out)
topo_out = outputs_function.maker.fgraph.toposort()
assert any(isinstance(t.op, cls) for t in topo_out)
# Check that the shape produced agrees with the actual shape.
numeric_outputs = outputs_function(*numeric_inputs)
numeric_shapes = shapes_function(*numeric_inputs)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论