masahi opened a new pull request #6383: URL: https://github.com/apache/incubator-tvm/pull/6383
This adds support for relay/topi where op to take scalar values as input. This use case came up in hummingbird project. See https://github.com/microsoft/hummingbird/issues/232#issuecomment-682262544 I think this is a reasonable feature to add. The output shape is the same as condition's shape. Condition's shape has to be 1D. We could allow condition shape to be arbitrary when x and y are scalar, but I didn't make that change. please review @zhiics @siju-samuel @junrushao1994 ---------------------------------------------------------------- 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: [email protected]
