Runner¶
Runner.
Contains functionality and classes needed for running an AutoDist-transformed graph with TensorFlow.
-
class
WrappedSession
(cluster, graph_item, remapper)[source]¶ Bases:
tensorflow.python.client.session.Session
Wraps tf.Session.
This allows us to use our transformed Graph and our Remapper on our cluster.