def run(self, fetches, feed_dict=None, options=None, run_metadata=None):
"""See base class."""
if self.should_stop():
raise RuntimeError('Run called even after should_stop requested.')
actual_fetches = {'caller': fetches}
run_context = session_run_hook.SessionRunContext(
original_args=session_run_hook.SessionRunArgs(fetches, feed_dict),
session=self._sess)
options = options or config_pb2.RunOptions()
feed_dict = self._call_hook_before_run(run_context, actual_fetches,
feed_dict, options)
# Do session run.
run_metadata = run_metadata or config_pb2.RunMetadata()
outputs = _WrappedSession.run(self,
fetches=actual_fetches,
feed_dict=feed_dict,
options=options,
run_metadata=run_metadata)
for hook in self._hooks:
hook.after_run(
run_context,
session_run_hook.SessionRunValues(
results=outputs[hook] if hook in outputs else None,
options=options,
run_metadata=run_metadata))
self._should_stop = self._should_stop or run_context.stop_requested
return outputs['caller']
def run_hook_with_indices(self, sweep_hook, row_indices, col_indices):
with self.test_session() as sess:
# Before run.
run_context = session_run_hook.SessionRunContext(
original_args=None, session=sess)
sess_run_args = sweep_hook.before_run(run_context)
feed_dict = {
self._input_row_indices_ph: row_indices,
self._input_col_indices_ph: col_indices
}
# Run.
run_results = sess.run(sess_run_args.fetches, feed_dict=feed_dict)
run_values = session_run_hook.SessionRunValues(
results=run_results, options=None, run_metadata=None)
# After run.
sweep_hook.after_run(run_context, run_values)
