| /external/tensorflow/tensorflow/contrib/eager/python/examples/l2hmc/ |
| D | l2hmc_test.py | 63 step_fn=step): argument
70 _, samples = step_fn(dynamics, optimizer, samples)
76 step_fn=step, argument
86 loss, samples = step_fn(dynamics, optimizer, samples)
210 step_fn = tfe.defun(step) if defun else step
218 step_fn=step_fn)
227 step_fn=step_fn,
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 1006 def step_fn(step_context): function
1008 return step_fn
1039 def step_fn(step_context): function
1041 return step_fn
1072 def step_fn(step_context): function
1074 return step_fn
1118 def step_fn(step_context): function
1121 return step_fn
1154 def step_fn(step_context): function
1157 return step_fn
[all …]
|
| D | monitored_session.py | 748 def run_step_fn(self, step_fn): argument
779 step_fn_arguments = function_utils.fn_args(step_fn)
794 return self._sess.run_step_fn(step_fn, self._tf_sess(), run_with_hooks=None)
1159 def run_step_fn(self, step_fn, raw_session, run_with_hooks): argument
1164 return step_fn(_MonitoredSession.StepContext(raw_session, run_with_hooks))
1254 def run_step_fn(self, step_fn, raw_session, run_with_hooks): argument
1261 return self._sess.run_step_fn(step_fn, raw_session, run_with_hooks)
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | minimize_loss_test.py | 64 def step_fn(ctx, inputs): function
74 step_fn, iterator, iterations=2).run_op
155 def step_fn(ctx, inputs): function
165 step_fn, iterator, iterations=1).run_op
221 def step_fn(ctx, inputs): function
234 step_fn, iterator, iterations=1).run_op
316 def step_fn(ctx, inputs): function
326 step_fn, iterator, iterations=1).run_op
401 def step_fn(output_context, inputs): function
432 step_fn, iterator, iterations=2,
|
| D | step_fn.py | 97 def step_fn(ctx, inputs): function
113 step_fn, self._iterator, self._iterations_per_step)
|
| D | single_loss_example.py | 21 from tensorflow.contrib.distribute.python import step_fn
46 single_loss_step = step_fn.StandardSingleLossStep(
|
| D | metrics_v1_test.py | 100 def step_fn(ctx, inputs): function
107 step_fn, iterator, iterations=distribution.extended.steps_per_run)
|
| D | BUILD | 286 name = "step_fn",
287 srcs = ["step_fn.py"],
468 ":step_fn",
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed.py | 218 def step_fn(ctx, inputs): function
264 return step_fn
326 step_fn = _make_step_fn(model, ModeKeys.TRAIN, current_strategy, out_labels)
347 step_fn, iterator, iterations=steps_per_run,
488 step_fn = _make_step_fn(model, ModeKeys.TEST, current_strategy, out_labels)
500 step_fn, iterator, iterations=1,
629 step_fn = _make_step_fn(model, ModeKeys.PREDICT, current_strategy, out_labels)
643 step_fn, iterator, iterations=1,
|
| /external/tensorflow/tensorflow/contrib/distribute/ |
| D | __init__.py | 32 from tensorflow.contrib.distribute.python.step_fn import *
|
| D | BUILD | 33 "//tensorflow/contrib/distribute/python:step_fn",
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.-monitored-session.pbtxt | 28 argspec: "args=[\'self\', \'step_fn\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.train.-singular-monitored-session.pbtxt | 32 argspec: "args=[\'self\', \'step_fn\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/python/debug/wrappers/ |
| D | framework.py | 609 def run_step_fn(self, step_fn): argument
610 return step_fn(
|