[GitHub] [tvm] masahi commented on issue #8412: [Topi] Allow relay.nn.dense support arbitrary number dimensions

2021-10-14 Thread GitBox
masahi commented on issue #8412: URL: https://github.com/apache/tvm/issues/8412#issuecomment-943180691 I think a good way to add N-D input support to `topi.dense` is to do `topi.reshape` on the input before doing dense compute, and inside dense schedules we do `compute_inline(...)` on the

[GitHub] [tvm] masahi commented on issue #8412: [Topi] Allow relay.nn.dense support arbitrary number dimensions

2021-10-14 Thread GitBox
masahi commented on issue #8412: URL: https://github.com/apache/tvm/issues/8412#issuecomment-943168765 See https://discuss.tvm.apache.org/t/relay-nn-does-relay-nn-dense-supports-multi-dimensional-input/10343/7. I realized that supporting more than 2D input for `dense` is extremely