You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2021/02/03 17:42:55 UTC

[GitHub] [tvm] mbrookhart opened a new issue #7402: Argwhere not returning all values?

mbrookhart opened a new issue #7402:
URL: https://github.com/apache/tvm/issues/7402


   @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 


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] mbrookhart commented on issue #7402: Argwhere not returning all values?

Posted by GitBox <gi...@apache.org>.
mbrookhart commented on issue #7402:
URL: https://github.com/apache/tvm/issues/7402#issuecomment-772694101


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


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] mbrookhart closed issue #7402: Argwhere not returning all values?

Posted by GitBox <gi...@apache.org>.
mbrookhart closed issue #7402:
URL: https://github.com/apache/tvm/issues/7402


   


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org