Commitf08b18e

David So

authored and

Mesh TensorFlow Team

committed

Remove unnecessary name and cwise in squared relu.

PiperOrigin-RevId: 396177118

1 parent739fb09 commitf08b18eCopy full SHA for f08b18e

File tree

-2

lines changed

-2

lines changed

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -1921,9 +1921,9 @@ def relu(x, name="relu"):`
`1921`	`1921`	`returncwise(tf.nn.relu, [x],name=name,grad_function=_relu_grad)`
`1922`	`1922`
`1923`	`1923`
`1924`		`-defsquared_relu(x,name="squared_relu"):`
	`1924`	`+defsquared_relu(x):`
`1925`	`1925`	`"""Squared ReLU from Primer paper (TODO(davidso):Link when released)."""`
`1926`		`-returncwise(lambdat:square(relu(t)), [x],name=name)`
	`1926`	`+returnsquare(relu(x))`
`1927`	`1927`
`1928`	`1928`
`1929`	`1929`	`defleaky_relu(x,alpha=0.2,name="leaky_relu"):`

Comments

(0)