Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
[TOPI] Update softmax compute and CPU schedule #3680
This change improves performance for softmax by simplifying the computation and writing a schedule that supports better parallelization.
Schedule: Currently, the schedule only parallelizes the
The following results are with a tensor of shape