Searched refs:_swish_shape (Results 1 – 1 of 1) sorted by relevance
339 def _swish_shape(op): function344 @function.Defun(shape_func=_swish_shape, func_name="swish_grad", noinline=True)361 shape_func=_swish_shape,