Add a link to the Primer paper

Mesh TensorFlow Team · Mesh TensorFlow Team · commit 52a2332c3bb0 · 2021-10-06T08:16:20.000-07:00
PiperOrigin-RevId: 401248102
diff --git a/mesh_tensorflow/ops.py b/mesh_tensorflow/ops.py
@@ -1922,7 +1922,7 @@ def relu(x, name="relu"):
 
 
 def squared_relu(x):
-  """Squared ReLU from Primer paper (TODO(davidso):Link when released)."""
+  """Squared ReLU from Primer paper: https://arxiv.org/abs/2109.08668."""
   return square(relu(x))
 
 
