c6daa24e3cac10e3488d5a2eadfd6134de55e677,keras/backend/theano_backend.py,,batch_normalization,#Any#Any#Any#Any#Any#Any#,386

Before Change

    normed = T.nnet.bn.batch_normalization(x, gamma, beta, mean,
                                           sqrt(var) + epsilon,
                                           mode="high_mem")
    return normed


// SHAPE OPERATIONS

After Change

def batch_normalization(x, mean, var, beta, gamma, epsilon=0.0001):
    """Apply batch normalization on x given mean, var, beta and gamma.
    """
    if theano.config.device.startswith("cuda") or theano.config.device.startswith("gpu"):
        try:
            return theano.sandbox.cuda.dnn.dnn_batch_normalization_test(x, gamma, beta, mean, var,
                                                                        "spatial", epsilon)
        except AttributeError:
            pass
    return T.nnet.bn.batch_normalization(x, gamma, beta, mean, sqrt(var) + epsilon,
                                         mode="high_mem")

Instances