• Ricardo Vieira's avatar
    Fix Blockwise infer shape from core Op · 6dd61726
    Ricardo Vieira 提交于
    Sometimes `_create_dummy_core_node` can create a multi-node graph, where the root inputs are not `node.inputs`. Then infer_shape may bypass the intermediate nodes. This was the case with Subtensor, which introduces `ScalarFromTensor` nodes, but ignores them in the shape graph (for a cleaner graph)
    6dd61726
test_blockwise.py 20.9 KB