clean incorrect relu test code

No substantial code change, will merge when passing.

This fixes the ReLU problem in #1355 . The MVN layer is more complex though.

The time benchmark test has been flaky for a while - I took the opportunity to relax it a little bit and added a variable so we can adjust it more easily.