0f37436eae7485aa586231f85b8fc6b2ad4b78ca,horovod/tensorflow/elastic.py,TensorFlowKerasState,__init__,#TensorFlowKerasState#Any#Any#Any#,102

Before Change

        def broadcast_object_with_session(obj):
            return broadcast_object(obj, session=backend.get_session())

        broadcast_object_fn = broadcast_object if not backend or _executing_eagerly() else broadcast_object_with_session

        super(TensorFlowKerasState, self).__init__(bcast_object=broadcast_object_fn,
                                                   get_rank=rank,
                                                   **kwargs)

After Change

            // For TensorFlow v1, we need to reuse the broadcast op to prevent incrementing the uids
            bcast_op = broadcast_variables(_global_variables(), root_rank=0)
            self._bcast_model = lambda: self.backend.get_session().run(bcast_op)
            bcast_object = broadcast_object_fn(session=self.backend.get_session())

        super(TensorFlowKerasState, self).__init__(bcast_object=bcast_object,
                                                   get_rank=rank,
                                                   **kwargs)

Instances