altanh opened a new pull request #8307: URL: https://github.com/apache/tvm/pull/8307
New gradients: - `cast_like` - `not_equal` - `strided_slice` - `one_hot` Also simplified `log_softmax` gradient. For `strided_slice`, I wasn't sure how to add support for the recently introduced `axes` argument (which if I understand correctly aims to circumvent limitations in the type system for dynamic shape inference), so I just added a check. In my mind, `strided_slice` (as opposed to `dyn.strided_slice`) should be used when everything is concrete, but I think now it allows for limited shape dynamism which atm isn't a good idea for training anyway. cc @masahi for confirmation -- 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]
