Argwhere not returning all values?

[mbrookhart]

@zhiics @masahi

I'm running into an odd issue with arghwere when I run this test:

def test_full_argwhere():
    x = relay.const(1)
    full = relay.full(x, [128])
    y = relay.argwhere(full)
    mod = tvm.IRModule()
    mod["main"] = relay.Function([], y)
    expected = np.argwhere(np.ones(128))
    check_result([], mod, expected, flatten=True)

TVM argwhere is only returning the first value, while numpy argwhere is effectively doing range. Any ideas why the tvm op is failing?

actual = array([0], dtype=int32)
desired = array([  0,   1,   2,   3,   4,   5,   6,   7,   8,   9,  10,  11,  12,
        13,  14,  15,  16,  17,  18,  19,  20,..., 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116,
       117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127])

cc @jwfromm

[mbrookhart]

Nevermind, this is a bug in something else I was doing.