[dtensor][math_ops] fixing math_ops single dim strategies to consider…