* refine gather test

* add masked_fill backward test

* add mean module backward test

* rewrite module to test

* refien

* add layernorm backward test

* refine math_ops tests

Co-authored-by: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com>