(input_tensor, scale)
| 223 | swish_scale = array_ops.constant(1.0) |
| 224 | |
| 225 | def _swish(input_tensor, scale): |
| 226 | custom = op_hint.OpHint("cool_activation") |
| 227 | input_tensor, scale = custom.add_inputs(input_tensor, scale) |
| 228 | output = math_ops.sigmoid(input_tensor) * input_tensor * scale |
| 229 | output, = custom.add_outputs(output) |
| 230 | return output |
| 231 | |
| 232 | output = array_ops.identity( |
| 233 | _swish(image, swish_scale), name="ModelOutput") |
nothing calls this directly
no test coverage detected