004b323e6fc19ba4ff96bcd48d8518d35a7c5e8c,test/conversion/test_load_batchnorm.py,TestBatchNorm,setUp,#TestBatchNorm#,18

Before Change

    def setUp(self):
         
        self.model = keras.models.Sequential()
        self.model = keras.models.Sequential()
        self.epsilon = 10**(-6)
        self.gamma = np.array([2.0, 3.0]) 
        self.beta = np.array([4.0, 5.0])

After Change

                            [self.keras_model.layers[0].input],
                            self.keras_model.layers[0].get_output(train=False),
                            allow_input_downcast=True) 
        self.keras_output_fprop_func = theano.function(
                        [self.keras_model.layers[0].input],
                        self.keras_model.layers[-1].get_output(train=False),
                        allow_input_downcast=True)

        grad = theano.grad(theano.tensor.sum(
                   self.keras_model.layers[-1].get_output(train=False)[:,0]),
                   self.keras_model.layers[0].input)

Instances