Understanding code organization: where is `log_softmax` really implemented?

2

You’ll find the CPU implementation here.

PS: rgrep is often a useful tool to find the corresponding function.
The GitHub search function is often also quite helpful.

1 Like