2fe1f942de95b3fe11acb4881c8f18f65d73bd8e,deepexplain/tensorflow/methods.py,,deepexplain_grad,#Any#Any#,109

Before Change

    cases = OrderedDict(
        (
            (tf.equal(mode, flag), lambda: method_class.nonlinearity_grad_override(op, grad))
            for method, (method_class, flag) in attribution_methods.items()
        )
    )
    return tf.case(cases, default=lambda: grad * grad_activation(op.name)(op.inputs[0]))

After Change

        input = op.inputs[0]
        return grad * grad_activation(op.name)(input)

    cases = OrderedDict({
        tf.equal(mode, 0): (lambda: DummyZero.nonlinearity_grad_override(op, grad)),
        tf.equal(mode, 1): (lambda: Saliency.nonlinearity_grad_override(op, grad)),
        tf.equal(mode, 2): (lambda: GradientXInput.nonlinearity_grad_override(op, grad)),
        tf.equal(mode, 3): (lambda: IntegratedGradients.nonlinearity_grad_override(op, grad)),
        tf.equal(mode, 4): (lambda: EpsilonLRP.nonlinearity_grad_override(op, grad)),
    })
    return tf.case(cases, default=default, exclusive=True)

Instances