MoisesHer opened a new pull request #18104: URL: https://github.com/apache/incubator-mxnet/pull/18104
## Description ## This PR includes broadcast_add into fused operators (still to complete) ## Checklist ## ### Essentials ### Please feel free to remove inapplicable items for your PR. - [ ] Changes are complete (i.e. I finished coding on this PR): Does not work in case input0_dims =! input1_dims => need to expand_dims of smaller input (in progress) - [ ] All changes have test coverage: - Unit tests are added for small changes to verify correctness (e.g. adding a new operator) Still to include - [x] Code is well-documented: - [x] To the best of my knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change ### Changes ### - [x] Created a new list of fused compatible ops (InputsOnly) for broadcast ops. For now only added broadcast_add - [x] Generation of code for broadcast ops was included. ## Comments ## TODO: - Expand the dimension of inputs when input0_dims =! input1_dims - Include test ---------------------------------------------------------------- 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]
