3e4541881bc659ec4ff59383df917aa319be7cae,test/test_tensorflow_keras.py,TfKerasTests,test_load_model_broadcast,#TfKerasTests#,209

Before Change

            K.clear_session()
            with self.session(config=self.config) as sess:
                K.set_session(sess)

                weight = np.random.random((1, 3))

                if hvd.rank() == 0:

After Change

        with temppath() as fname:
            with self.session(config=self.config) as sess:
                K.set_session(sess)

                model = create_model()

                x = np.random.random((1, 3))
                y = np.random.random((1, 3, 3))

Instances