Runner

Runner.

Contains functionality and classes needed for running an AutoDist-transformed graph with TensorFlow.

get_default_session_config()[source]

Create a default session config.

get_default_run_options()[source]

Create a default run option.

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.

run(fetches, feed_dict=None, options=None, run_metadata=None)[source]

Wrapped Session.run.