(ctx, grad_output)
| 78 | |
| 79 | @staticmethod |
| 80 | def backward(ctx, grad_output): |
| 81 | out, = ctx.saved_tensors |
| 82 | |
| 83 | grad_input, grad_bias = FusedLeakyReLUFunctionBackward.apply( |
| 84 | grad_output, out, ctx.bias, ctx.negative_slope, ctx.scale |
| 85 | ) |
| 86 | |
| 87 | if not ctx.bias: |
| 88 | grad_bias = None |
| 89 | |
| 90 | return grad_input, grad_bias, None, None |
| 91 | |
| 92 | |
| 93 | class FusedLeakyReLU(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected