Remove unnecessary name and cwise in squared relu.

PiperOrigin-RevId: 396177118

Author: David So

mesh_tensorflow/ops.py

@@ -1921,9 +1921,9 @@ def relu(x, name="relu"):
   return cwise(tf.nn.relu, [x], name=name, grad_function=_relu_grad)
 
 
-def squared_relu(x, name="squared_relu"):
+def squared_relu(x):
   """Squared ReLU from Primer paper (TODO(davidso):Link when released)."""
-  return cwise(lambda t: square(relu(t)), [x], name=name)
+  return square(relu(x))
 
 
 def leaky_relu(x, alpha=0.2, name="leaky_relu"):