I would like to work on this issue:
It is a simple function, but I think it needs to be implement at a very low level, since none of the primitives in the current API can address the issue.
Looking at the implementation of the
relu functions should be helpful. But I can’t quite located their low-level implementations in the repo. It would be great if someone could point them out to me.
Any other pointers are also much appreciated.